From 48e9b1745ae6a53fe4c5fc2d54022373c3f27922 Mon Sep 17 00:00:00 2001 From: Kayla Meyer Date: Fri, 1 Dec 2023 11:30:06 +0100 Subject: [PATCH] bug-fix-zarr chunking --- deeprvat/deeprvat/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deeprvat/deeprvat/train.py b/deeprvat/deeprvat/train.py index fdc184a4..9f9257bb 100644 --- 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