Skip to content

Commit 96608d9

Browse files
committed
update docstrings to reference zarr.codecs.numcodecs
1 parent c6ce404 commit 96608d9

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

src/zarr/metadata/migrate_v3.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -252,7 +252,7 @@ def _convert_compressor(
252252
return GzipCodec(level=compressor.level)
253253

254254
case _:
255-
# If possible, find matching numcodecs.zarr3 codec
255+
# If possible, find matching zarr.codecs.numcodecs codec
256256
compressor_codec = _find_numcodecs_zarr3(compressor)
257257

258258
if not isinstance(compressor_codec, BytesBytesCodec):
@@ -262,7 +262,7 @@ def _convert_compressor(
262262

263263

264264
def _find_numcodecs_zarr3(numcodecs_codec: numcodecs.abc.Codec) -> Codec:
265-
"""Find matching numcodecs.zarr3 codec (if it exists)"""
265+
"""Find matching zarr.codecs.numcodecs codec (if it exists)"""
266266

267267
numcodec_name = f"numcodecs.{numcodecs_codec.codec_id}"
268268
numcodec_dict = {
@@ -274,7 +274,7 @@ def _find_numcodecs_zarr3(numcodecs_codec: numcodecs.abc.Codec) -> Codec:
274274
codec_v3 = get_codec_class(numcodec_name)
275275
except KeyError as exc:
276276
raise ValueError(
277-
f"Couldn't find corresponding numcodecs.zarr3 codec for {numcodecs_codec.codec_id}"
277+
f"Couldn't find corresponding zarr.codecs.numcodecs codec for {numcodecs_codec.codec_id}"
278278
) from exc
279279

280280
return codec_v3.from_dict(numcodec_dict)

tests/test_cli/test_migrate_v3.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
from zarr.codecs.blosc import BloscCodec
1414
from zarr.codecs.bytes import BytesCodec
1515
from zarr.codecs.gzip import GzipCodec
16-
from zarr.codecs.numcodecs._codecs import LZMA, Delta
16+
from zarr.codecs.numcodecs import LZMA, Delta
1717
from zarr.codecs.transpose import TransposeCodec
1818
from zarr.codecs.zstd import ZstdCodec
1919
from zarr.core.array import Array
@@ -506,7 +506,8 @@ def test_migrate_unknown_codec(local_store: LocalStore) -> None:
506506
assert result.exit_code == 1
507507
assert isinstance(result.exception, ValueError)
508508
assert (
509-
str(result.exception) == "Couldn't find corresponding numcodecs.zarr3 codec for categorize"
509+
str(result.exception)
510+
== "Couldn't find corresponding zarr.codecs.numcodecs codec for categorize"
510511
)
511512

512513

0 commit comments

Comments
 (0)