fix codec initialization for zarr3

Author: tasansal

src/mdio/api/convenience.py

@@ -5,7 +5,7 @@
 from typing import TYPE_CHECKING
 
 import zarr
-from numcodecs import Blosc
+from numcodecs.zarr3 import Blosc
 from tqdm.auto import tqdm
 
 from mdio import MDIOReader
@@ -134,8 +134,8 @@ def create_rechunk_plan(
     metadata_arrs = []
     data_arrs = []
 
-    header_compressor = Blosc("zstd")
-    trace_compressor = Blosc("zstd") if compressors is None else compressors
+    header_compressor = Blosc(cname="zstd")
+    trace_compressor = Blosc(cname="zstd") if compressors is None else compressors
 
     for chunks, suffix in zip(chunks_list, suffix_list, strict=True):
         norm_chunks = tuple(min(chunk, size) for chunk, size in zip(chunks, source.shape, strict=True))

src/mdio/core/factory.py

@@ -25,7 +25,7 @@
 from typing import Any
 
 import zarr
-from numcodecs import Blosc
+from numcodecs.zarr3 import Blosc
 from numpy.typing import DTypeLike
 from zarr import Group
 from zarr import open_group
@@ -160,7 +160,7 @@ def create_empty(
             name=f"{variable.name}_trace_headers",
             shape=config.grid.shape[:-1],  # Same spatial shape as data
             chunks=variable.chunks[:-1],  # Same spatial chunks as data
-            compressors=Blosc("zstd"),
+            compressors=Blosc(cname="zstd"),
             dtype=header_dtype,
         )
 

src/mdio/schemas/v1/dataset_serializer.py

@@ -2,7 +2,7 @@
 
 import numpy as np
 from dask import array as dask_array
-from numcodecs import Blosc as nc_Blosc
+from numcodecs.zarr3 import Blosc as nc_Blosc
 from xarray import DataArray as xr_DataArray
 from xarray import Dataset as xr_Dataset
 
@@ -127,7 +127,7 @@ def _convert_compressor(
 
     if isinstance(compressor, mdio_Blosc):
         return nc_Blosc(
-            cname=compressor.algorithm.value,
+            cname=compressor.algorithm,
             clevel=compressor.level,
             shuffle=compressor.shuffle.value,
             blocksize=compressor.blocksize if compressor.blocksize > 0 else 0,