rslearn 0.0.2__tar.gz → 0.0.3__tar.gz

{rslearn-0.0.2/rslearn.egg-info → rslearn-0.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: rslearn
-Version: 0.0.2
+Version: 0.0.3
 Summary: A library for developing remote sensing datasets and models
 Author-email: Favyen Bastani <favyenb@allenai.org>, Yawen Zhang <yawenz@allenai.org>, Patrick Beukema <patrickb@allenai.org>, Henry Herzog <henryh@allenai.org>, Piper Wolters <piperw@allenai.org>
 License:                                  Apache License
@@ -205,41 +205,57 @@ License:                                  Apache License
            See the License for the specific language governing permissions and
            limitations under the License.
         
-Requires-Python: >=3.10
+Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: boto3>=1.35
+Requires-Dist: boto3>=1.39
 Requires-Dist: class_registry>=2.1
 Requires-Dist: fiona>=1.10
-Requires-Dist: fsspec[gcs,s3]>=2024.10
-Requires-Dist: lightning[pytorch-extra]>=2.4
-Requires-Dist: Pillow>=11.0
+Requires-Dist: fsspec==2025.3.0
+Requires-Dist: jsonargparse>=4.35.0
+Requires-Dist: lightning>=2.5.1.post0
+Requires-Dist: Pillow>=11.3
 Requires-Dist: pyproj>=3.7
 Requires-Dist: python-dateutil>=2.9
 Requires-Dist: pytimeparse>=1.1
 Requires-Dist: rasterio>=1.4
-Requires-Dist: shapely>=2.0
-Requires-Dist: torch>=2.5
-Requires-Dist: torchvision>=0.20
-Requires-Dist: tqdm>=4.66
-Requires-Dist: universal_pathlib>=0.2.5
+Requires-Dist: shapely>=2.1
+Requires-Dist: torch>=2.7.0
+Requires-Dist: torchvision>=0.22.0
+Requires-Dist: tqdm>=4.67
+Requires-Dist: universal_pathlib>=0.2.6
 Provides-Extra: extra
-Requires-Dist: accelerate>=1.0; extra == "extra"
-Requires-Dist: earthengine-api>=0.1; extra == "extra"
+Requires-Dist: accelerate>=1.10; extra == "extra"
+Requires-Dist: cdsapi>=0.7.6; extra == "extra"
+Requires-Dist: earthdaily[platform]>=1.0.0; extra == "extra"
+Requires-Dist: earthengine-api>=1.6.3; extra == "extra"
 Requires-Dist: einops>=0.8; extra == "extra"
-Requires-Dist: gcsfs>=2024.10; extra == "extra"
-Requires-Dist: google-cloud-bigquery>=2.18; extra == "extra"
+Requires-Dist: gcsfs==2025.3.0; extra == "extra"
+Requires-Dist: google-cloud-bigquery>=3.35; extra == "extra"
 Requires-Dist: google-cloud-storage>=2.18; extra == "extra"
-Requires-Dist: interrogate>=1.7; extra == "extra"
-Requires-Dist: osmium>=3.7; extra == "extra"
-Requires-Dist: planet>=2.10; extra == "extra"
+Requires-Dist: netCDF4>=1.7.2; extra == "extra"
+Requires-Dist: osmium>=4.0.2; extra == "extra"
+Requires-Dist: planet>=3.1; extra == "extra"
+Requires-Dist: planetary_computer>=1.0; extra == "extra"
 Requires-Dist: pycocotools>=2.0; extra == "extra"
-Requires-Dist: rtree>=1.2; extra == "extra"
-Requires-Dist: s3fs>=2024.10.0; extra == "extra"
+Requires-Dist: pystac_client>=0.9; extra == "extra"
+Requires-Dist: rtree>=1.4; extra == "extra"
+Requires-Dist: s3fs==2025.3.0; extra == "extra"
 Requires-Dist: satlaspretrain_models>=0.3; extra == "extra"
-Requires-Dist: scipy>=1.13; extra == "extra"
-Requires-Dist: transformers>=4.45; extra == "extra"
-Requires-Dist: wandb>=0.17; extra == "extra"
+Requires-Dist: scipy>=1.16; extra == "extra"
+Requires-Dist: terratorch>=1.0.2; extra == "extra"
+Requires-Dist: transformers>=4.55; extra == "extra"
+Requires-Dist: wandb>=0.21; extra == "extra"
+Provides-Extra: dev
+Requires-Dist: interrogate>=1.7.0; extra == "dev"
+Requires-Dist: mypy<2,>=1.17.1; extra == "dev"
+Requires-Dist: pre-commit>=4.3.0; extra == "dev"
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest_httpserver; extra == "dev"
+Requires-Dist: ruff>=0.12.9; extra == "dev"
+Requires-Dist: pytest-dotenv; extra == "dev"
+Requires-Dist: pytest-xdist; extra == "dev"
+Dynamic: license-file
 
 Overview
 --------
@@ -259,10 +275,11 @@ rslearn helps with:
 
 
 Quick links:
-- [CoreConcepts](CoreConcepts.md) summarizes key concepts in rslearn, including
+- [CoreConcepts](docs/CoreConcepts.md) summarizes key concepts in rslearn, including
   datasets, windows, layers, and data sources.
-- [Examples](Examples.md) contains more examples, including customizing different
+- [Examples](docs/Examples.md) contains more examples, including customizing different
   stages of rslearn with additional code.
+- [DatasetConfig](docs/DatasetConfig.md) documents the dataset configuration file.
 
 
 Setup
@@ -270,9 +287,33 @@ Setup
 
 rslearn requires Python 3.10+ (Python 3.12 is recommended).
 
-    git clone https://github.com/allenai/rslearn.git
-    cd rslearn
-    pip install .[extra]
+```
+git clone https://github.com/allenai/rslearn.git
+cd rslearn
+pip install .[extra]
+```
+
+
+Supported Data Sources
+----------------------
+
+rslearn supports ingesting raster and vector data from the following data sources. Even
+if you don't plan to train models within rslearn, you can still use it to easily
+download, crop, and re-project data based on spatiotemporal rectangles (windows) that
+you define. See [Examples](docs/Examples.md) and [DatasetConfig](docs/DatasetConfig.md)
+for how to setup these data sources.
+
+- Sentinel-1
+- Sentinel-2 L1C and L2A
+- Landsat 8/9 OLI-TIRS
+- National Agriculture Imagery Program
+- OpenStreetMap
+- Xyz (Slippy) Tiles (e.g., Mapbox tiles)
+- Planet Labs (PlanetScope, SkySat)
+- ESA WorldCover 2021
+
+rslearn can also be used to easily mosaic, crop, and re-project any sets of local
+raster and vector files you may have.
 
 
 Example Usage
@@ -286,28 +327,25 @@ Let's start by defining a region of interest and obtaining Sentinel-2 images. Cr
 directory `/path/to/dataset` and corresponding configuration file at
 `/path/to/dataset/config.json` as follows:
 
-    {
-        "layers": {
-            "sentinel2": {
-                "type": "raster",
-                "band_sets": [{
-                    "dtype": "uint8",
-                    "bands": ["R", "G", "B"]
-                }],
-                "data_source": {
-                    "name": "rslearn.data_sources.gcp_public_data.Sentinel2",
-                    "index_cache_dir": "cache/sentinel2/",
-                    "max_time_delta": "1d",
-                    "sort_by": "cloud_cover",
-                    "use_rtree_index": false
-                }
+```json
+{
+    "layers": {
+        "sentinel2": {
+            "type": "raster",
+            "band_sets": [{
+                "dtype": "uint8",
+                "bands": ["R", "G", "B"]
+            }],
+            "data_source": {
+                "name": "rslearn.data_sources.gcp_public_data.Sentinel2",
+                "index_cache_dir": "cache/sentinel2/",
+                "sort_by": "cloud_cover",
+                "use_rtree_index": false
             }
-        },
-        "tile_store": {
-            "name": "file",
-            "root_dir": "tiles"
         }
     }
+}
+```
 
 Here, we have initialized an empty dataset and defined a raster layer called
 `sentinel2`. Because it specifies a data source, it will be populated automatically. In
@@ -319,8 +357,10 @@ choosing the scenes with minimal cloud cover.
 Next, let's create our spatiotemporal windows. These will correspond to training
 examples.
 
-    export DATASET_PATH=/path/to/dataset
-    rslearn dataset add_windows --root $DATASET_PATH --group default --utm --resolution 10 --grid_size 128 --src_crs EPSG:4326 --box=-122.6901,47.2079,-121.4955,47.9403 --start 2024-06-01T00:00:00+00:00 --end 2024-08-01T00:00:00+00:00 --name seattle
+```
+export DATASET_PATH=/path/to/dataset
+rslearn dataset add_windows --root $DATASET_PATH --group default --utm --resolution 10 --grid_size 128 --src_crs EPSG:4326 --box=-122.6901,47.2079,-121.4955,47.9403 --start 2024-06-01T00:00:00+00:00 --end 2024-08-01T00:00:00+00:00 --name seattle
+```
 
 This creates windows along a 128x128 grid in the specified projection (i.e.,
 appropriate UTM zone for the location with 10 m/pixel resolution) covering the
@@ -332,9 +372,11 @@ We can now obtain the Sentinel-2 images by running prepare, ingest, and material
 * Ingest: retrieve those items. This step populates the `tiles` directory within the dataset.
 * Materialize: crop/mosaic the items to align with the windows. This populates the `layers` folder in each window directory.
 
-    rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
-    rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
-    rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
+rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
+rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
+rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
 
 For ingestion, you may need to reduce the number of workers depending on the available
 memory on your system.
@@ -342,32 +384,36 @@ memory on your system.
 You should now be able to open the GeoTIFF images. Let's find the window that
 corresponds to downtown Seattle:
 
-    import shapely
-    from rslearn.const import WGS84_PROJECTION
-    from rslearn.dataset import Dataset
-    from rslearn.utils import Projection, STGeometry
-    from upath import UPath
-
-    # Define longitude and latitude for downtown Seattle.
-    downtown_seattle = shapely.Point(-122.333, 47.606)
-
-    # Iterate over the windows and find the closest one.
-    dataset = Dataset(path=UPath("/path/to/dataset"))
-    best_window_name = None
-    best_distance = None
-    for window in dataset.load_windows(workers=32):
-        shp = window.get_geometry().to_projection(WGS84_PROJECTION).shp
-        distance = shp.distance(downtown_seattle)
-        if best_distance is None or distance < best_distance:
-            best_window_name = window.name
-            best_distance = distance
-
-    print(best_window_name)
+```python
+import shapely
+from rslearn.const import WGS84_PROJECTION
+from rslearn.dataset import Dataset
+from rslearn.utils import Projection, STGeometry
+from upath import UPath
+
+# Define longitude and latitude for downtown Seattle.
+downtown_seattle = shapely.Point(-122.333, 47.606)
+
+# Iterate over the windows and find the closest one.
+dataset = Dataset(path=UPath("/path/to/dataset"))
+best_window_name = None
+best_distance = None
+for window in dataset.load_windows(workers=32):
+    shp = window.get_geometry().to_projection(WGS84_PROJECTION).shp
+    distance = shp.distance(downtown_seattle)
+    if best_distance is None or distance < best_distance:
+        best_window_name = window.name
+        best_distance = distance
+
+print(best_window_name)
+```
 
 It should be `seattle_54912_-527360`, so let's open it in qgis (or your favorite GIS
 software):
 
-    qgis $DATASET_PATH/windows/default/seattle_54912_-527360/layers/sentinel2/R_G_B/geotiff.tif
+```
+qgis $DATASET_PATH/windows/default/seattle_54912_-527360/layers/sentinel2/R_G_B/geotiff.tif
+```
 
 
 ### Adding Land Cover Labels
@@ -377,152 +423,164 @@ the ESA WorldCover land cover map as labels.
 
 Start by downloading the WorldCover data from https://worldcover2021.esa.int
 
-    wget https://worldcover2021.esa.int/data/archive/ESA_WorldCover_10m_2021_v200_60deg_macrotile_N30W180.zip
-    mkdir world_cover_tifs
-    unzip ESA_WorldCover_10m_2021_v200_60deg_macrotile_N30W180.zip -d world_cover_tifs/
+```
+wget https://worldcover2021.esa.int/data/archive/ESA_WorldCover_10m_2021_v200_60deg_macrotile_N30W180.zip
+mkdir world_cover_tifs
+unzip ESA_WorldCover_10m_2021_v200_60deg_macrotile_N30W180.zip -d world_cover_tifs/
+```
 
 It would require some work to write a script to re-project and crop these GeoTIFFs so
 that they align with the windows we have previously defined (and the Sentinel-2 images
 we have already ingested). We can use the LocalFiles data source to have rslearn
 automate this process. Update the dataset `config.json` with a new layer:
 
-    "layers": {
-        "sentinel2": {
-            ...
-        },
-        "worldcover": {
-            "type": "raster",
-            "band_sets": [{
-                "dtype": "uint8",
-                "bands": ["B1"]
-            }],
-            "resampling_method": "nearest",
-            "data_source": {
-                "name": "rslearn.data_sources.local_files.LocalFiles",
-                "src_dir": "file:///path/to/world_cover_tifs/"
-            }
-        }
+```json
+"layers": {
+    "sentinel2": {
+        ...
     },
-    ...
+    "worldcover": {
+        "type": "raster",
+        "band_sets": [{
+            "dtype": "uint8",
+            "bands": ["B1"]
+        }],
+        "resampling_method": "nearest",
+        "data_source": {
+            "name": "rslearn.data_sources.local_files.LocalFiles",
+            "src_dir": "file:///path/to/world_cover_tifs/"
+        }
+    }
+},
+...
+```
 
 Repeat the materialize process so we populate the data for this new layer:
 
-    rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
-    rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
-    rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
+rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
+rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
+rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
 
 We can visualize both the GeoTIFFs together in qgis:
 
-    qgis $DATASET_PATH/windows/default/seattle_54912_-527360/layers/*/*/geotiff.tif
+```
+qgis $DATASET_PATH/windows/default/seattle_54912_-527360/layers/*/*/geotiff.tif
+```
 
 
 ### Training a Model
 
 Create a model configuration file `land_cover_model.yaml`:
 
+```yaml
+model:
+  class_path: rslearn.train.lightning_module.RslearnLightningModule
+  init_args:
+    # This part defines the model architecture.
+    # Essentially we apply the SatlasPretrain Sentinel-2 backbone with a UNet decoder
+    # that terminates at a segmentation prediction head.
+    # The backbone outputs four feature maps at different scales, and the UNet uses
+    # these to compute a feature map at the input scale.
+    # Finally the segmentation head applies per-pixel softmax to compute the land
+    # cover class.
     model:
-      class_path: rslearn.train.lightning_module.RslearnLightningModule
+      class_path: rslearn.models.singletask.SingleTaskModel
       init_args:
-        # This part defines the model architecture.
-        # Essentially we apply the SatlasPretrain Sentinel-2 backbone with a UNet decoder
-        # that terminates at a segmentation prediction head.
-        # The backbone outputs four feature maps at different scales, and the UNet uses
-        # these to compute a feature map at the input scale.
-        # Finally the segmentation head applies per-pixel softmax to compute the land
-        # cover class.
-        model:
-          class_path: rslearn.models.singletask.SingleTaskModel
-          init_args:
-            encoder:
-              - class_path: rslearn.models.satlaspretrain.SatlasPretrain
-                init_args:
-                  model_identifier: "Sentinel2_SwinB_SI_RGB"
-            decoder:
-              - class_path: rslearn.models.unet.UNetDecoder
-                init_args:
-                  in_channels: [[4, 128], [8, 256], [16, 512], [32, 1024]]
-                  # We use 101 classes because the WorldCover classes are 10, 20, 30, 40
-                  # 50, 60, 70, 80, 90, 95, 100.
-                  # We could process the GeoTIFFs to collapse them to 0-10 (the 11 actual
-                  # classes) but the model will quickly learn that the intermediate
-                  # values are never used.
-                  out_channels: 101
-                  conv_layers_per_resolution: 2
-              - class_path: rslearn.train.tasks.segmentation.SegmentationHead
-        # Remaining parameters in RslearnLightningModule define different aspects of the
-        # training process like initial learning rate.
-        lr: 0.0001
-    data:
-      class_path: rslearn.train.data_module.RslearnDataModule
+        encoder:
+          - class_path: rslearn.models.satlaspretrain.SatlasPretrain
+            init_args:
+              model_identifier: "Sentinel2_SwinB_SI_RGB"
+        decoder:
+          - class_path: rslearn.models.unet.UNetDecoder
+            init_args:
+              in_channels: [[4, 128], [8, 256], [16, 512], [32, 1024]]
+              # We use 101 classes because the WorldCover classes are 10, 20, 30, 40
+              # 50, 60, 70, 80, 90, 95, 100.
+              # We could process the GeoTIFFs to collapse them to 0-10 (the 11 actual
+              # classes) but the model will quickly learn that the intermediate
+              # values are never used.
+              out_channels: 101
+              conv_layers_per_resolution: 2
+          - class_path: rslearn.train.tasks.segmentation.SegmentationHead
+    # Remaining parameters in RslearnLightningModule define different aspects of the
+    # training process like initial learning rate.
+    lr: 0.0001
+data:
+  class_path: rslearn.train.data_module.RslearnDataModule
+  init_args:
+    # Replace this with the dataset path.
+    path: /path/to/dataset/
+    # This defines the layers that should be read for each window.
+    # The key ("image" / "targets") is what the data will be called in the model,
+    # while the layers option specifies which layers will be read.
+    inputs:
+      image:
+        data_type: "raster"
+        layers: ["sentinel2"]
+        bands: ["R", "G", "B"]
+        passthrough: true
+      targets:
+        data_type: "raster"
+        layers: ["worldcover"]
+        bands: ["B1"]
+        is_target: true
+    task:
+      # Train for semantic segmentation.
+      # The remap option is only used when visualizing outputs during testing.
+      class_path: rslearn.train.tasks.segmentation.SegmentationTask
       init_args:
-        # Replace this with the dataset path.
-        path: /path/to/dataset/
-        # This defines the layers that should be read for each window.
-        # The key ("image" / "targets") is what the data will be called in the model,
-        # while the layers option specifies which layers will be read.
-        inputs:
-          image:
-            data_type: "raster"
-            layers: ["sentinel2"]
-            bands: ["R", "G", "B"]
-            passthrough: true
-          targets:
-            data_type: "raster"
-            layers: ["worldcover"]
-            bands: ["B1"]
-            is_target: true
-        task:
-          # Train for semantic segmentation.
-          # The remap option is only used when visualizing outputs during testing.
-          class_path: rslearn.train.tasks.segmentation.SegmentationTask
+        num_classes: 101
+        remap_values: [[0, 1], [0, 255]]
+    batch_size: 8
+    num_workers: 32
+    # These define different options for different phases/splits, like training,
+    # validation, and testing.
+    # Here we use the same transform across splits except training where we add a
+    # flipping augmentation.
+    # For now we are using the same windows for training and validation.
+    default_config:
+      transforms:
+        - class_path: rslearn.train.transforms.normalize.Normalize
           init_args:
-            num_classes: 101
-            remap_values: [[0, 1], [0, 255]]
-        batch_size: 8
-        num_workers: 32
-        # These define different options for different phases/splits, like training,
-        # validation, and testing.
-        # Here we use the same transform across splits except training where we add a
-        # flipping augmentation.
-        # For now we are using the same windows for training and validation.
-        default_config:
-          transforms:
-            - class_path: rslearn.train.transforms.normalize.Normalize
-              init_args:
-                mean: 0
-                std: 255
-        train_config:
-          transforms:
-            - class_path: rslearn.train.transforms.normalize.Normalize
-              init_args:
-                mean: 0
-                std: 255
-            - class_path: rslearn.train.transforms.flip.Flip
-              init_args:
-                image_selectors: ["image", "target/classes", "target/valid"]
-          groups: ["default"]
-        val_config:
-          groups: ["default"]
-        test_config:
-          groups: ["default"]
-        predict_config:
-          groups: ["predict"]
-          load_all_patches: true
-          skip_targets: true
-          patch_size: 512
-    trainer:
-      max_epochs: 10
-      callbacks:
-        - class_path: lightning.pytorch.callbacks.ModelCheckpoint
+            mean: 0
+            std: 255
+    train_config:
+      transforms:
+        - class_path: rslearn.train.transforms.normalize.Normalize
+          init_args:
+            mean: 0
+            std: 255
+        - class_path: rslearn.train.transforms.flip.Flip
           init_args:
-            save_top_k: 1
-            save_last: true
-            monitor: val_accuracy
-            mode: max
+            image_selectors: ["image", "target/classes", "target/valid"]
+      groups: ["default"]
+    val_config:
+      groups: ["default"]
+    test_config:
+      groups: ["default"]
+    predict_config:
+      groups: ["predict"]
+      load_all_patches: true
+      skip_targets: true
+      patch_size: 512
+trainer:
+  max_epochs: 10
+  callbacks:
+    - class_path: lightning.pytorch.callbacks.ModelCheckpoint
+      init_args:
+        save_top_k: 1
+        save_last: true
+        monitor: val_accuracy
+        mode: max
+```
 
 Now we can train the model:
 
-    rslearn model fit --config land_cover_model.yaml
+```
+rslearn model fit --config land_cover_model.yaml
+```
 
 
 ### Apply the Model
@@ -533,22 +591,26 @@ windows along a grid, we just create one big window. This is because we are just
 to run the prediction over the whole window rather than use different windows as
 different training examples.
 
-    rslearn dataset add_windows --root $DATASET_PATH --group predict --utm --resolution 10 --src_crs EPSG:4326 --box=-122.712,45.477,-122.621,45.549 --start 2024-06-01T00:00:00+00:00 --end 2024-08-01T00:00:00+00:00 --name portland
-    rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
-    rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
-    rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
+rslearn dataset add_windows --root $DATASET_PATH --group predict --utm --resolution 10 --src_crs EPSG:4326 --box=-122.712,45.477,-122.621,45.549 --start 2024-06-01T00:00:00+00:00 --end 2024-08-01T00:00:00+00:00 --name portland
+rslearn dataset prepare --root $DATASET_PATH --workers 32 --batch-size 8
+rslearn dataset ingest --root $DATASET_PATH --workers 32 --no-use-initial-job --jobs-per-process 1
+rslearn dataset materialize --root $DATASET_PATH --workers 32 --no-use-initial-job
+```
 
 We also need to add an RslearnPredictionWriter to the trainer callbacks in the model
 configuration file, as it will handle writing the outputs from the model to a GeoTIFF.
 
-    trainer:
-      callbacks:
-        - class_path: lightning.pytorch.callbacks.ModelCheckpoint
-          ...
-        - class_path: rslearn.train.prediction_writer.RslearnWriter
-          init_args:
-            path: /path/to/dataset/
-            output_layer: output
+```yaml
+trainer:
+  callbacks:
+    - class_path: lightning.pytorch.callbacks.ModelCheckpoint
+      ...
+    - class_path: rslearn.train.prediction_writer.RslearnWriter
+      init_args:
+        path: /path/to/dataset/
+        output_layer: output
+```
 
 Because of our `predict_config`, when we run `model predict` it will apply the model on
 windows in the "predict" group, which is where we added the Portland window.
@@ -556,39 +618,46 @@ windows in the "predict" group, which is where we added the Portland window.
 And it will be written in a new output_layer called "output". But we have to update the
 dataset configuration so it specifies the layer:
 
-
-    "layers": {
-        "sentinel2": {
-            ...
-        },
-        "worldcover": {
-            ...
-        },
-        "output": {
-            "type": "raster",
-            "band_sets": [{
-                "dtype": "uint8",
-                "bands": ["output"]
-            }]
-        }
+```json
+"layers": {
+    "sentinel2": {
+        ...
+    },
+    "worldcover": {
+        ...
     },
+    "output": {
+        "type": "raster",
+        "band_sets": [{
+            "dtype": "uint8",
+            "bands": ["output"]
+        }]
+    }
+},
+```
 
 Now we can apply the model:
 
-    # Find model checkpoint in lightning_logs dir.
-    ls lightning_logs/*/checkpoints/last.ckpt
-    rslearn model predict --config land_cover_model.yaml --ckpt_path lightning_logs/version_0/checkpoints/last.ckpt
+```
+# Find model checkpoint in lightning_logs dir.
+ls lightning_logs/*/checkpoints/last.ckpt
+rslearn model predict --config land_cover_model.yaml --ckpt_path lightning_logs/version_0/checkpoints/last.ckpt
+```
 
 And visualize the Sentinel-2 image and output in qgis:
 
-    qgis $DATASET_PATH/windows/predict/portland/layers/*/*/geotiff.tif
+```
+qgis $DATASET_PATH/windows/predict/portland/layers/*/*/geotiff.tif
+```
 
 
 ### Defining Train and Validation Splits
 
 We can visualize the logged metrics using Tensorboard:
 
-    tensorboard --logdir=lightning_logs/
+```
+tensorboard --logdir=lightning_logs/
+```
 
 However, because our training and validation data are identical, the validation metrics
 are not meaningful.
@@ -602,57 +671,61 @@ We will use the second approach. The script below sets a "split" key in the opti
 dict (which is stored in each window's `metadata.json` file) to "train" or "val"
 based on the SHA-256 hash of the window name.
 
-    import hashlib
-    import tqdm
-    from rslearn.dataset import Dataset, Window
-    from upath import UPath
-
-    ds_path = UPath("/path/to/dataset/")
-    dataset = Dataset(ds_path)
-    windows = dataset.load_windows(show_progress=True, workers=32)
-    for window in tqdm.tqdm(windows):
-        if hashlib.sha256(window.name.encode()).hexdigest()[0] in ["0", "1"]:
-            split = "val"
-        else:
-            split = "train"
-        if "split" in window.options and window.options["split"] == split:
-            continue
-        window.options["split"] = split
-        window.save()
+```python
+import hashlib
+import tqdm
+from rslearn.dataset import Dataset, Window
+from upath import UPath
+
+ds_path = UPath("/path/to/dataset/")
+dataset = Dataset(ds_path)
+windows = dataset.load_windows(show_progress=True, workers=32)
+for window in tqdm.tqdm(windows):
+    if hashlib.sha256(window.name.encode()).hexdigest()[0] in ["0", "1"]:
+        split = "val"
+    else:
+        split = "train"
+    if "split" in window.options and window.options["split"] == split:
+        continue
+    window.options["split"] = split
+    window.save()
+```
 
 Now we can update the model configuration file to use these splits:
 
-    default_config:
-      transforms:
-        - class_path: rslearn.train.transforms.normalize.Normalize
-          init_args:
-            mean: 0
-            std: 255
-    train_config:
-      transforms:
-        - class_path: rslearn.train.transforms.normalize.Normalize
-          init_args:
-            mean: 0
-            std: 255
-        - class_path: rslearn.train.transforms.flip.Flip
-          init_args:
-            image_selectors: ["image", "target/classes", "target/valid"]
-      groups: ["default"]
-      tags:
-        split: train
-    val_config:
-      groups: ["default"]
-      tags:
-        split: val
-    test_config:
-      groups: ["default"]
-      tags:
-        split: val
-    predict_config:
-      groups: ["predict"]
-      load_all_patches: true
-      skip_targets: true
-      patch_size: 512
+```yaml
+default_config:
+  transforms:
+    - class_path: rslearn.train.transforms.normalize.Normalize
+      init_args:
+        mean: 0
+        std: 255
+train_config:
+  transforms:
+    - class_path: rslearn.train.transforms.normalize.Normalize
+      init_args:
+        mean: 0
+        std: 255
+    - class_path: rslearn.train.transforms.flip.Flip
+      init_args:
+        image_selectors: ["image", "target/classes", "target/valid"]
+  groups: ["default"]
+  tags:
+    split: train
+val_config:
+  groups: ["default"]
+  tags:
+    split: val
+test_config:
+  groups: ["default"]
+  tags:
+    split: val
+predict_config:
+  groups: ["predict"]
+  load_all_patches: true
+  skip_targets: true
+  patch_size: 512
+```
 
 The `tags` option that we are adding here tells rslearn to only load windows with a
 matching key and value in the window options.
@@ -660,13 +733,17 @@ matching key and value in the window options.
 Previously when we run `model fit`, it should show the same number of windows for
 training and validation:
 
-    got 4752 examples in split train
-    got 4752 examples in split val
+```
+got 4752 examples in split train
+got 4752 examples in split val
+```
 
 With the updates, it should show different numbers like this:
 
-    got 4167 examples in split train
-    got 585 examples in split val
+```
+got 4167 examples in split train
+got 585 examples in split val
+```
 
 
 ### Visualizing with `model test`