Fix save masks bug (#318)

opengeos · Sep 21, 2024 · 856a7a0 · 856a7a0
1 parent 3e3a310
commit 856a7a0
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/samgeo/samgeo.py b/samgeo/samgeo.py
@@ -222,11 +222,10 @@ def generate(
         self.masks = masks  # Store the masks as a list of dictionaries
         self.batch = False
 
-        if output is not None:
-            # Save the masks to the output path. The output is either a binary mask or a mask of objects with unique values.
-            self.save_masks(
-                output, foreground, unique, erosion_kernel, mask_multiplier, **kwargs
-            )
+        # Save the masks to the output path. The output is either a binary mask or a mask of objects with unique values.
+        self.save_masks(
+            output, foreground, unique, erosion_kernel, mask_multiplier, **kwargs
+        )
 
     def save_masks(
         self,