Bugfix: from array with data chunks.

hyperspy · Jan 17, 2024 · 995818c · 995818c
1 parent 4301190
commit 995818c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rsciio/_hierarchical.py b/rsciio/_hierarchical.py
@@ -267,8 +267,9 @@ 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