Fix chunks for connected components calculation

AlexeyPechnikov · Sep 13, 2024 · adbd18d · adbd18d
1 parent 4c084d3
commit adbd18d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pygmtsar/pygmtsar/Stack_unwrap.py b/pygmtsar/pygmtsar/Stack_unwrap.py
@@ -570,7 +570,7 @@ def find_mode(array):
             return values[max_count_index] if counts.size > 0 else np.nan
 
         # Apply the function along the 'pair' dimension
-        maincomps =  xr.apply_ufunc(find_mode, conncomp, input_core_dims=[['y', 'x']],
+        maincomps =  xr.apply_ufunc(find_mode, conncomp.chunk(dict(y=-1, x=-1)), input_core_dims=[['y', 'x']],
                                     vectorize=True, dask='parallelized', output_dtypes=[int])
         return data.where(conncomp==maincomps)