minor changes

Signed-off-by: João Lucas de Sousa Almeida <[email protected]>
IBM · Jan 7, 2025 · d6486f3 · d6486f3
1 parent d58ff1b
commit d6486f3
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/terratorch/datasets/sen1floods11.py b/terratorch/datasets/sen1floods11.py
@@ -23,7 +23,8 @@
 
 class Sen1Floods11NonGeo(NonGeoDataset):
     """NonGeo dataset implementation for sen1floods11."""
-        all_band_names = (
+
+    all_band_names = (
             "COASTAL_AEROSOL",
             "BLUE",
             "GREEN",
@@ -37,15 +38,11 @@ class Sen1Floods11NonGeo(NonGeoDataset):
             "CIRRUS",
             "SWIR_1",
             "SWIR_2",
-        )
-
+    )
     rgb_bands = ("RED", "GREEN", "BLUE")
-
     BAND_SETS = {"all": all_band_names, "rgb": rgb_bands}
-
     num_classes = 2
     splits = {"train": "train", "val": "valid", "test": "test"}
-
     data_dir = "v1.1/data/flood_events/HandLabeled/S2Hand"
     label_dir = "v1.1/data/flood_events/HandLabeled/LabelHand"
     split_dir = "v1.1/splits/flood_handlabeled"