bug-fix-zarr chunking

PMBio · Dec 1, 2023 · 96ccf8f · 96ccf8f
1 parent 48e9b17
commit 96ccf8f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/deeprvat/deeprvat/train.py b/deeprvat/deeprvat/train.py
@@ -309,7 +309,7 @@ def __init__(
                     pheno_data["input_tensor_zarr"],
                     self.zarr_root,
                     name=pheno,
-                    chunks=(self.chunksize, None, None, None, None),
+                    chunks=(self.chunksize, None, None, None),
                     compressor=Blosc(clevel=1),
                 )
                 pheno_data["input_tensor_zarr"] = self.zarr_root[pheno]
@@ -321,7 +321,7 @@ def __init__(
                 zarr.copy(
                     pheno_data["input_tensor_zarr"],
                     zarr.DirectoryStore(tensor_path),
-                    chunks=(self.chunksize, None, None, None, None),
+                    chunks=(self.chunksize, None, None, None),
                     compressor=Blosc(clevel=1),
                 )
                 pheno_data["input_tensor_zarr"] = zarr.open(tensor_path)