diff --git a/torchgeo/datamodules/substation.py b/torchgeo/datamodules/substation.py index c066f0c4cd..ce95ded249 100644 --- 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,