.rechunk()-Issue with dask 2024.11.2 #11544

# Load reconstructions
Reconstructions = [dask_image.imread.imread(os.path.join(f, '*rec*.png')) for f in Data.Folder]
# Save to zarr
Data['OutputNameZarr'] = [f + '.zarr' for f in Data.Folder]
for c, row in Data.iterrows():
        Reconstructions[c].rechunk(chunks='auto').to_zarr(row['OutputNameZarr'])
# Load zarr files back in
Reconstructions = [dask.array.from_zarr(file) for file in Data['OutputNameZarr']]
# Select one reconstruction and crop a small region
whichsample = 12
center=(2000,2000,1250)
sidelength=333
Crop = Reconstructions[whichsample][center[2] - (sidelength // 2):center[2] + ((1+sidelength) // 2), center[1] - (sidelength // 2):center[1] + ((1+sidelength) // 2), center[0] - (sidelength // 2):center[0] + ((1+sidelength) // 2)]
Crop.rechunk(chunks='auto').to_zarr(OutputNameCrop)

---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
Cell In[43], line 2
      1 whichsample = 2
----> 2 Cutout = cutout(whichsample,
      3                 center=(2000,2000,1250),
      4                 sidelength=333,
      5                 verbose=True)

Cell In[42], line 39, in cutout(whichsample, center, sidelength, verbose)
     35     Crop = Reconstructions[whichsample][center[2] - (sidelength // 2):center[2] + ((1+sidelength) // 2),
     36                            center[1] - (sidelength // 2):center[1] + ((1+sidelength) // 2),
     37                            center[0] - (sidelength // 2):center[0] + ((1+sidelength) // 2)]
     38     # Save to disk
---> 39     Crop.rechunk().to_zarr(OutputNameCrop)
     40 if verbose:
     41     for d, direction in enumerate(directions):

File [~\AppData\Local\anaconda3\envs\laminitis\Lib\site-packages\dask\array\core.py:3015](http://localhost:8888/~/AppData/Local/anaconda3/envs/laminitis/Lib/site-packages/dask/array/core.py#line=3014), in Array.to_zarr(self, *args, **kwargs)
   3004 def to_zarr(self, *args, **kwargs):
   3005     """Save array to the zarr storage format
   3006 
   3007     See https://zarr.readthedocs.io for details about the format.
   (...)
   3013     dask.array.to_zarr : equivalent function
   3014     """
-> 3015     return to_zarr(self, *args, **kwargs)

File ~\AppData\Local\anaconda3\envs\laminitis\Lib\site-packages\dask\array\core.py:3875, in to_zarr(arr, url, component, storage_options, overwrite, region, compute, return_stored, **kwargs)
   3872     raise ValueError("Cannot use `region` keyword when url is not a `zarr.Array`.")
   3874 if not _check_regular_chunks(arr.chunks):
-> 3875     raise ValueError(
   3876         "Attempt to save array to zarr with irregular "
   3877         "chunking, please call `arr.rechunk(...)` first."
   3878     )
   3880 storage_options = storage_options or {}
   3882 if storage_options:

ValueError: Attempt to save array to zarr with irregular chunking, please call `arr.rechunk(...)` first.