Fix: Dense OOC Concat With Aligned Elements Memory Issue (#1169)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
scverse · Oct 10, 2023 · 08ef471 · 08ef471
1 parent c66a463
commit 08ef471
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/anndata/experimental/merge.py b/anndata/experimental/merge.py
@@ -172,7 +172,10 @@ def write_concat_dense(
     """
     import dask.array as da
 
-    darrays = (da.from_array(a, chunks="auto") for a in arrays)
+    darrays = (
+        da.from_array(a, chunks="auto" if a.chunks is None else a.chunks)
+        for a in arrays
+    )
 
     res = da.concatenate(
         [