2121# libraries in the pre-commit mypy environment.
2222
2323
24- @gpu_test # type: ignore[misc ,unused-ignore]
24+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
2525@pytest .mark .parametrize ("store" , ["local" , "memory" ], indirect = ["store" ])
2626@pytest .mark .parametrize (
2727 "checksum" ,
@@ -64,7 +64,7 @@ def test_nvcomp_zstd(store: Store, checksum: bool, selection: tuple[slice, slice
6464 cp .testing .assert_array_equal (expected [:, :], a [:, :])
6565
6666
67- @gpu_test # type: ignore[misc ,unused-ignore]
67+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
6868@pytest .mark .parametrize ("host_encode" , [True , False ])
6969def test_gpu_codec_compatibility (host_encode : bool ) -> None :
7070 # Ensure that the we can decode CPU-encoded data with the GPU
@@ -116,7 +116,7 @@ def gpu_context() -> Iterator[None]:
116116 xp .testing .assert_array_equal (result , read_data )
117117
118118
119- @gpu_test # type: ignore[misc ,unused-ignore]
119+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
120120def test_invalid_raises () -> None :
121121 with pytest .raises (ValueError ):
122122 NvcompZstdCodec (level = 100 , checksum = False )
@@ -128,7 +128,7 @@ def test_invalid_raises() -> None:
128128 NvcompZstdCodec (checksum = "False" ) # type: ignore[arg-type,unused-ignore]
129129
130130
131- @gpu_test # type: ignore[misc ,unused-ignore]
131+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
132132def test_uses_default_codec () -> None :
133133 with zarr .config .enable_gpu ():
134134 a = zarr .create_array (
@@ -141,7 +141,7 @@ def test_uses_default_codec() -> None:
141141 assert isinstance (a .metadata .codecs [- 1 ], NvcompZstdCodec )
142142
143143
144- @gpu_test # type: ignore[misc ,unused-ignore]
144+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
145145def test_nvcomp_from_dict () -> None :
146146 config : dict [str , JSON ] = {
147147 "name" : "zstd" ,
@@ -155,31 +155,31 @@ def test_nvcomp_from_dict() -> None:
155155 assert codec .checksum is False
156156
157157
158- @gpu_test # type: ignore[misc ,unused-ignore]
158+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
159159def test_compute_encoded_chunk_size () -> None :
160160 codec = NvcompZstdCodec (level = 0 , checksum = False )
161161 with pytest .raises (NotImplementedError ):
162162 codec .compute_encoded_size (
163163 _input_byte_length = 0 ,
164164 _chunk_spec = ArraySpec (
165165 shape = (10 , 10 ),
166- dtype = zarr .core .dtype .npy . int . Int32 (),
166+ dtype = zarr .core .dtype .Int32 (),
167167 fill_value = 0 ,
168168 config = ArrayConfig (order = "C" , write_empty_chunks = False ),
169169 prototype = buffer_prototype ,
170170 ),
171171 )
172172
173173
174- @gpu_test # type: ignore[misc ,unused-ignore]
174+ @gpu_test # type: ignore[untyped-decorator ,unused-ignore]
175175async def test_nvcomp_zstd_encode_none () -> None :
176176 codec = NvcompZstdCodec (level = 0 , checksum = False )
177177 chunks_and_specs = [
178178 (
179179 None ,
180180 ArraySpec (
181181 shape = (10 , 10 ),
182- dtype = zarr .core .dtype .npy . int . Int32 (),
182+ dtype = zarr .core .dtype .Int32 (),
183183 fill_value = 0 ,
184184 config = ArrayConfig (order = "C" , write_empty_chunks = False ),
185185 prototype = buffer_prototype ,
0 commit comments