set kwargs type

Author: brokkoli71

src/zarr/codecs/blosc.py

@@ -4,7 +4,7 @@
 from dataclasses import dataclass, replace
 from enum import Enum
 from functools import cached_property
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 import numcodecs
 from numcodecs.blosc import Blosc
@@ -106,7 +106,7 @@ def __init__(
         clevel: int = 5,
         shuffle: BloscShuffle | str | None = None,
         blocksize: int = 0,
-        **kwargs: Any,
+        **kwargs: JSON,
     ) -> None:
         reject_must_understand_metadata(kwargs, "`blosc` codec configuration")
 

src/zarr/codecs/bytes.py

@@ -3,7 +3,7 @@
 import sys
 from dataclasses import dataclass, replace
 from enum import Enum
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 import numpy as np
 
@@ -45,7 +45,7 @@ def __init__(
         self,
         *,
         endian: Endian | str | None = default_system_endian,
-        **kwargs: Any,
+        **kwargs: JSON,
     ) -> None:
         reject_must_understand_metadata(kwargs, "`bytes` codec configuration")
 

src/zarr/codecs/gzip.py

@@ -2,7 +2,7 @@
 
 import asyncio
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 from numcodecs.gzip import GZip
 
@@ -34,7 +34,7 @@ class GzipCodec(BytesBytesCodec):
 
     level: int = 5
 
-    def __init__(self, *, level: int = 5, **kwargs: Any) -> None:
+    def __init__(self, *, level: int = 5, **kwargs: JSON) -> None:
         reject_must_understand_metadata(kwargs, "`gzip` codec configuration")
         level_parsed = parse_gzip_level(level)
 

src/zarr/codecs/sharding.py

@@ -344,7 +344,7 @@ def __init__(
         codecs: Iterable[Codec | dict[str, JSON]] = (BytesCodec(),),
         index_codecs: Iterable[Codec | dict[str, JSON]] = (BytesCodec(), Crc32cCodec()),
         index_location: ShardingCodecIndexLocation | str = ShardingCodecIndexLocation.end,
-        **kwargs: Any,
+        **kwargs: JSON,
     ) -> None:
         reject_must_understand_metadata(kwargs, "`sharding` codec configuration")
 

src/zarr/codecs/transpose.py

@@ -37,7 +37,7 @@ class TransposeCodec(ArrayArrayCodec):
 
     order: tuple[int, ...]
 
-    def __init__(self, *, order: ChunkCoordsLike, **kwargs: Any) -> None:
+    def __init__(self, *, order: ChunkCoordsLike, **kwargs: JSON) -> None:
         reject_must_understand_metadata(kwargs, "`transpose` codec configuration")
         order_parsed = parse_transpose_order(order)
 

src/zarr/codecs/vlen_utf8.py

@@ -1,7 +1,7 @@
 from __future__ import annotations
 
 from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 from warnings import warn
 
 import numpy as np
@@ -26,7 +26,7 @@
 
 @dataclass(frozen=True)
 class VLenUTF8Codec(ArrayBytesCodec):
-    def __init__(self, **kwargs: Any) -> None:
+    def __init__(self, **kwargs: JSON) -> None:
         reject_must_understand_metadata(kwargs, "`vlen-utf8` codec configuration")
         warn(
             "The codec `vlen-utf8` is currently not part in the Zarr format 3 specification. It "
@@ -82,7 +82,7 @@ def compute_encoded_size(self, input_byte_length: int, _chunk_spec: ArraySpec) -
 
 @dataclass(frozen=True)
 class VLenBytesCodec(ArrayBytesCodec):
-    def __init__(self, **kwargs: Any) -> None:
+    def __init__(self, **kwargs: JSON) -> None:
         reject_must_understand_metadata(kwargs, "`vlen-bytes` codec configuration")
         warn(
             "The codec `vlen-bytes` is currently not part in the Zarr format 3 specification. It "

src/zarr/codecs/zstd.py

@@ -3,7 +3,7 @@
 import asyncio
 from dataclasses import dataclass
 from functools import cached_property
-from typing import TYPE_CHECKING, Any
+from typing import TYPE_CHECKING
 
 import numcodecs
 from numcodecs.zstd import Zstd
@@ -42,7 +42,7 @@ class ZstdCodec(BytesBytesCodec):
     level: int = 0
     checksum: bool = False
 
-    def __init__(self, *, level: int = 0, checksum: bool = False, **kwargs: Any) -> None:
+    def __init__(self, *, level: int = 0, checksum: bool = False, **kwargs: JSON) -> None:
         reject_must_understand_metadata(kwargs, "`zstd` codec configuration")
 
         # numcodecs 0.13.0 introduces the checksum attribute for the zstd codec

tests/test_metadata/test_v3.py

@@ -413,7 +413,7 @@ def test_dtypes(dtype_str: str) -> None:
         assert dt.byte_count is None
 
 
-def default_metadata_dict(**kwargs: Any) -> dict[str, Any]:
+def default_metadata_dict(**kwargs: JSON) -> dict[str, JSON]:
     d = {
         "zarr_format": 3,
         "node_type": "array",
@@ -509,7 +509,7 @@ def default_metadata_dict(**kwargs: Any) -> dict[str, Any]:
     ],
 )
 def test_fail_on_invalid_metadata_key(
-    metadata_dict: dict[str, Any], is_valid: bool, fail_msg: str
+    metadata_dict: dict[str, JSON], is_valid: bool, fail_msg: str
 ) -> None:
     if is_valid:
         ArrayV3Metadata.from_dict(metadata_dict)