got rid of attributes

freemansw1 · Nov 13, 2024 · 6725463 · 6725463
1 parent 2a3cf62
commit 6725463
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tobac/segmentation/watershed_segmentation.py b/tobac/segmentation/watershed_segmentation.py
@@ -1249,7 +1249,7 @@ def segmentation(
 
     # create our output dataarray
     segmentation_out_data = (
-        field.astype("int64", casting="unsafe")
+        xr.zeros_like(field, dtype=int)
         .rename("segmentation_mask")
         .assign_attrs(threshold=threshold)
     )