Bugfix: from array with data chunks.

hyperspy · Jan 17, 2024 · 63b145b · 63b145b
1 parent 4301190
commit 63b145b
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/rsciio/_hierarchical.py b/rsciio/_hierarchical.py
@@ -267,8 +267,7 @@ def _read_array(group, dataset_key):
             # cast to a numpy array to avoid multiple calls to
             # _decode_chunk in zarr (or h5py)
             data = da.from_array(data, chunks=data.chunks)
-            shape = da.from_array(ragged_shape, chunks=ragged_shape.chunks)
-            shape = shape.rechunk(data.chunks)
+            shape = da.from_array(ragged_shape, chunks=data.chunks)  # same chunks as data
             data = da.apply_gufunc(unflatten_data, "(),()->()", data, shape)
         return data