Undo changes to the Task too

nasaharvest · Nov 8, 2023 · fbc503b · fbc503b
1 parent 2dce11c
commit fbc503b
Showing 1 changed file with 4 additions and 7 deletions.
diff --git a/cropharvest/datasets.py b/cropharvest/datasets.py
@@ -31,7 +31,7 @@
 
 @dataclass
 class Task:
-    bounding_box: Optional[Union[BBox, List[BBox]]] = None
+    bounding_box: Optional[BBox] = None
     target_label: Optional[str] = None
     balance_negative_crops: bool = False
     test_identifier: Optional[str] = None
@@ -47,12 +47,9 @@ def __post_init__(self):
                 )
 
         if self.bounding_box is None:
-            self.bounding_box = [
-                BBox(min_lat=-90, max_lat=90, min_lon=-180, max_lon=180, name="global")
-            ]
-        else:
-            if isinstance(self.bounding_box, BBox):
-                self.bounding_box = [self.bounding_box]
+            self.bounding_box = BBox(
+                min_lat=-90, max_lat=90, min_lon=-180, max_lon=180, name="global"
+            )
 
     @property
     def id(self) -> str: