Enhance dask arguments processing in nanconvolve2d_gaussian()

AlexeyPechnikov · Aug 26, 2024 · b9306e5 · b9306e5
1 parent a47826d
commit b9306e5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pygmtsar/pygmtsar/utils.py b/pygmtsar/pygmtsar/utils.py
@@ -88,8 +88,8 @@ def nanconvolve2d_gaussian_dask(data, weight, **kwargs):
             # use map_overlap with the custom function to handle both arrays
             return da.map_overlap(
                 nanconvolve2d_gaussian_dask_chunk,
-                *(da.broadcast_arrays(data, weight) if weight is not None else [data]),
-                depth=depth,
+                *([data, weight] if weight is not None else [data]),
+                depth={0: depth[0], 1: depth[1]},
                 boundary='none',
                 dtype=data.dtype,
                 meta=data._meta,