bug-fix-zarr chunking

PMBio · Dec 1, 2023 · 48e9b17 · 48e9b17
1 parent efa3e0d
commit 48e9b17
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deeprvat/deeprvat/train.py b/deeprvat/deeprvat/train.py
@@ -247,7 +247,7 @@ def make_dataset(
         zarr.save_array(
             input_tensor_out_file,
             input_tensor.numpy(),
-            chunks=(1000, None, None, None, None),
+            chunks=(1000, None, None, None),
             compressor=Blosc(clevel=compression_level),
         )
         del input_tensor