From bd8c8d13ecb37715ed1f1c659a5156cd93358728 Mon Sep 17 00:00:00 2001 From: Lurvelly Date: Sat, 15 Nov 2025 12:40:09 +0800 Subject: [PATCH] Change compressor to Zarr-3's new codec system. --- roboverse_learn/il/data2zarr_dp.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/roboverse_learn/il/data2zarr_dp.py b/roboverse_learn/il/data2zarr_dp.py index c43961b79..0bb6d1826 100644 --- a/roboverse_learn/il/data2zarr_dp.py +++ b/roboverse_learn/il/data2zarr_dp.py @@ -8,6 +8,7 @@ import numpy as np import torch import zarr +from zarr.codecs import BloscCodec, BloscShuffle from tqdm import tqdm try: @@ -86,7 +87,7 @@ def main(): zarr_meta = zarr_root.create_group("meta") # ZARR datasets will be created dynamically during the first batch write - compressor = zarr.Blosc(cname="zstd", clevel=3, shuffle=1) + compressor = BloscCodec(cname="zstd", clevel=3, shuffle="shuffle") # Batch processing settings batch_size = 100