Initialize chunks correctly with empty channels_3d

fmi-faim · Oct 9, 2024 · 9e92f81 · 9e92f81
1 parent 2c42b0d
commit 9e92f81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/metamorph_mda_parser/xarray.py b/src/metamorph_mda_parser/xarray.py
@@ -27,7 +27,7 @@ def _load_file(row, channels_3d: list[bool]):
     time = row['time']
     channel = row['channel']
 
-    chunks = (-1,) * (3 if channels_3d[channel] else 2)
+    chunks = (-1,) * (2 if not channels_3d or not channels_3d[channel] else 3)
     with imread(path, aszarr=True) as store:
         data = da.from_zarr(store, chunks=chunks)