stack dask array along last axis to avoid transpose

brainglobe · Mar 27, 2024 · c9283ed · c9283ed
1 parent a7c5e1b
commit c9283ed
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/examples/convert_tiffs_to_zarr.py b/examples/convert_tiffs_to_zarr.py
@@ -49,7 +49,6 @@
 
     # Read the tiff stack into a dask array
     da_arr = read_tiff_stack_with_dask(channel_dir)
-    da_arr = da_arr.transpose((2, 1, 0))
 
     # Add the dask array to the zarr group
     group.add_full_res_data(

diff --git a/src/stack_to_chunk/io_helpers.py b/src/stack_to_chunk/io_helpers.py
@@ -86,4 +86,4 @@ def read_tiff_stack_with_dask(path: Path) -> da.Array:
         da.from_delayed(delayed_reader, shape=shape, dtype=dtype)
         for delayed_reader in lazy_arrays
     ]
-    return da.stack(dask_arrays, axis=0)
+    return da.stack(dask_arrays, axis=-1)