changed bands datatype from datamodule

microsoft · Jan 8, 2025 · 173a915 · 173a915
1 parent d35b435
commit 173a915
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datamodules/substation.py b/torchgeo/datamodules/substation.py
@@ -24,7 +24,7 @@ def __init__(
         normalizing_factor: np.ndarray[Any, Any] | None = None,
         means: np.ndarray[Any, Any] | None = None,
         stds: np.ndarray[Any, Any] | None = None,
-        bands: int = 13,
+        bands: list[int] = [1,2,3],
         num_of_timepoints: int = 4,
         model_type: str = 'default',
         geo_transforms: Any | None = None,