chunk dense arrays in same chunk_size as sparse arrays

chanzuckerberg · Dec 20, 2024 · fa25f40 · fa25f40
1 parent f33e3c3
commit fa25f40
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/cellxgene_schema_cli/cellxgene_schema/utils.py b/cellxgene_schema_cli/cellxgene_schema/utils.py
@@ -134,7 +134,8 @@ def callback(func, elem_name: str, elem, iospec):
                 n_vars = elem.attrs.get("shape")[1]
                 return read_elem_as_dask(elem, chunks=(chunk_size, n_vars))
             elif iospec.encoding_type == "array" and len(elem.shape) > 1:
-                return read_elem_as_dask(elem)
+                n_vars = elem.shape[1]
+                return read_elem_as_dask(elem, chunks=(chunk_size, n_vars))
             else:
                 return func(elem)
         else: