minor change: for consistency the mask is cast to DataArray after the…

… if/else statement
Cloud-Drift · Dec 28, 2023 · e66db2d · e66db2d
1 parent df0ed7a
commit e66db2d
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/clouddrift/ragged.py b/clouddrift/ragged.py
@@ -830,9 +830,9 @@ def _mask_var(
         if len(var) == len(rowsize):
             mask = criterion(var)
         else:
-            mask = xr.DataArray(
-                data=apply_ragged(criterion, var, rowsize), dims=[dim_name]
-            ).astype(bool)
+            mask = apply_ragged(criterion, var, rowsize)
+
+        mask = xr.DataArray(data=mask, dims=[dim_name]).astype(bool)
 
         if not len(var) == len(mask):
             raise ValueError(