Skip to content

Commit 856a7a0

Browse files
authored
Fix save masks bug (#318)
1 parent 3e3a310 commit 856a7a0

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

samgeo/samgeo.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -222,11 +222,10 @@ def generate(
222222
self.masks = masks # Store the masks as a list of dictionaries
223223
self.batch = False
224224

225-
if output is not None:
226-
# Save the masks to the output path. The output is either a binary mask or a mask of objects with unique values.
227-
self.save_masks(
228-
output, foreground, unique, erosion_kernel, mask_multiplier, **kwargs
229-
)
225+
# Save the masks to the output path. The output is either a binary mask or a mask of objects with unique values.
226+
self.save_masks(
227+
output, foreground, unique, erosion_kernel, mask_multiplier, **kwargs
228+
)
230229

231230
def save_masks(
232231
self,

0 commit comments

Comments
 (0)