Use delayed for to_zarr to allow schedualer parameters

niaid · Jun 24, 2024 · 89ce9c0 · 89ce9c0
1 parent 3f3ba49
commit 89ce9c0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pytools/utils/zarr.py b/pytools/utils/zarr.py
@@ -34,7 +34,7 @@ def bin_shrink(img, shrink_dim=None):
     return img.astype(input_type)
 
 
-def build_pyramid(zarr_path, components: List[str], chunks=None, shrink=[-1, -2], overwrite=False):
+def build_pyramid(zarr_path, components: List[str], chunks=None, shrink=(-1, -2), overwrite=False):
     """
     Iteratively generate multiple resolutions of zarr arrays.
 
@@ -60,8 +60,9 @@ def build_pyramid(zarr_path, components: List[str], chunks=None, shrink=[-1, -2]
             compressor=arr.compressor,
             dimension_separator=arr._dimension_separator,
             filters=arr.filters,
+            compute=False,
             overwrite=overwrite,
             # dtype=arr.dtype.newbyteorder("="),
-        )
+        ).compute()
 
         source_image = da.from_zarr(zarr_path, component=output_component)