changed the datatype of bands to list[int] form int

microsoft · Jan 8, 2025 · d35b435 · d35b435
1 parent 7626f28
commit d35b435
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datasets/substation.py b/torchgeo/datasets/substation.py
@@ -46,7 +46,7 @@ class Substation(NonGeoDataset):
     def __init__(
         self,
         root: str,
-        bands: int,
+        bands: list[int],
         mask_2d: bool,
         timepoint_aggregation: str = 'concat',
         download: bool = False,