resolved comments

microsoft · Feb 27, 2025 · 6607100 · 6607100
1 parent efae346
commit 6607100
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 3 deletions.
diff --git a/tests/conf/substation.yaml b/tests/conf/substation.yaml
@@ -14,3 +14,6 @@ data:
     batch_size: 1
   dict_kwargs:
     root: 'tests/data/substation'
+    bands: [0, 1, 2]
+    num_of_timepoints: 4
+    timepoint_aggregation: 'median'
diff --git a/tests/datasets/test_substation.py b/tests/datasets/test_substation.py
@@ -4,7 +4,6 @@
 import os
 import shutil
 from pathlib import Path
-from typing import Any
 
 import matplotlib.pyplot as plt
 import pytest
@@ -202,4 +201,4 @@ def test_extract(self, tmp_path: Path) -> None:
             timepoint_aggregation='median',
             num_of_timepoints=4,
             root=tmp_path,
-        )
+        )
diff --git a/torchgeo/datamodules/substation.py b/torchgeo/datamodules/substation.py
@@ -69,7 +69,7 @@ def setup(self, stage: str) -> None:
             mask_2d=False,
             download=True,
             checksum=False,
-            **self.hparams,
+            **self.kwargs,
         )
 
         generator = torch.Generator().manual_seed(0)

diff --git a/torchgeo/datasets/substation.py b/torchgeo/datasets/substation.py
@@ -32,6 +32,7 @@ class Substation(NonGeoDataset):
     (i.e., revisits).
 
     Dataset Format:
+	
     * .npz file for each datapoint
 
     Dataset Features: