make numcodecs codecs backwards compatible

Author: d-v-b

src/zarr/codecs/blosc.py

@@ -170,6 +170,8 @@ def parse_shuffle(data: object) -> BloscShuffle:
         return data
     if data in BLOSC_SHUFFLE:
         return BloscShuffle[data]  # type: ignore[misc]
+    if data in range(len(BLOSC_SHUFFLE)):
+        return BloscShuffle.from_int(data)  # type: ignore[arg-type]
     raise TypeError(f"Value must be one of {BLOSC_SHUFFLE}. Got {data} instead.")
 
 

src/zarr/codecs/gzip.py

@@ -80,7 +80,7 @@ def to_json(self, zarr_format: ZarrFormat) -> GZipJSON_V2 | GZipJSON_V3:
         )  # pragma: no cover
 
     @classmethod
-    def _check_json_v2(cls, data: CodecJSON) -> TypeGuard[GZipJSON_V2]:
+    def _check_json_v2(cls, data: object) -> TypeGuard[GZipJSON_V2]:
         return (
             isinstance(data, Mapping)
             and set(data.keys()) == {"id", "level"}
@@ -89,7 +89,7 @@ def _check_json_v2(cls, data: CodecJSON) -> TypeGuard[GZipJSON_V2]:
         )
 
     @classmethod
-    def _check_json_v3(cls, data: CodecJSON) -> TypeGuard[GZipJSON_V3]:
+    def _check_json_v3(cls, data: object) -> TypeGuard[GZipJSON_V3]:
         return (
             isinstance(data, Mapping)
             and set(data.keys()) == {"name", "configuration"}

src/zarr/codecs/numcodecs/__init__.py

@@ -3,32 +3,32 @@
 from typing import Final
 
 from zarr.codecs.numcodecs._codecs import (
-    BZ2,
-    CRC32,
-    CRC32C,
-    LZ4,
-    LZMA,
-    ZFPY,
-    Adler32,
-    BitRound,
-    Blosc,
-    Fletcher32,
-    GZip,
-    JenkinsLookup3,
-    PackBits,
-    PCodec,
-    Quantize,
-    Shuffle,
-    Zlib,
-    Zstd,
     _NumcodecsArrayArrayCodec,
     _NumcodecsArrayBytesCodec,
     _NumcodecsBytesBytesCodec,
     _NumcodecsCodec,
 )
+from zarr.codecs.numcodecs.adler32 import Adler32
 from zarr.codecs.numcodecs.astype import AsType
+from zarr.codecs.numcodecs.bitround import BitRound
+from zarr.codecs.numcodecs.blosc import Blosc
+from zarr.codecs.numcodecs.bz2 import BZ2
+from zarr.codecs.numcodecs.crc32 import CRC32
+from zarr.codecs.numcodecs.crc32c import CRC32C
 from zarr.codecs.numcodecs.delta import Delta
 from zarr.codecs.numcodecs.fixed_scale_offset import FixedScaleOffset
+from zarr.codecs.numcodecs.fletcher32 import Fletcher32
+from zarr.codecs.numcodecs.gzip import GZip
+from zarr.codecs.numcodecs.jenkins_lookup3 import JenkinsLookup3
+from zarr.codecs.numcodecs.lz4 import LZ4
+from zarr.codecs.numcodecs.lzma import LZMA
+from zarr.codecs.numcodecs.packbits import PackBits
+from zarr.codecs.numcodecs.pcodec import PCodec
+from zarr.codecs.numcodecs.quantize import Quantize
+from zarr.codecs.numcodecs.shuffle import Shuffle
+from zarr.codecs.numcodecs.zfpy import ZFPY
+from zarr.codecs.numcodecs.zlib import Zlib
+from zarr.codecs.numcodecs.zstd import Zstd
 
 # This is a fixed dictionary of numcodecs codecs for which we have pre-made Zarr V3 wrappers
 NUMCODECS_WRAPPERS: Final[dict[str, type[_NumcodecsCodec]]] = {