55import numcodecs .abc
66
77import zarr
8+ from zarr import Array , Group
89from zarr .abc .codec import ArrayArrayCodec , BytesBytesCodec , Codec
910from zarr .abc .store import Store
1011from zarr .codecs .blosc import BloscCodec , BloscShuffle
1112from zarr .codecs .bytes import BytesCodec
1213from zarr .codecs .gzip import GzipCodec
1314from zarr .codecs .transpose import TransposeCodec
1415from zarr .codecs .zstd import ZstdCodec
15- from zarr .core .array import Array
1616from zarr .core .buffer .core import default_buffer_prototype
1717from zarr .core .chunk_key_encodings import V2ChunkKeyEncoding
1818from zarr .core .common import (
2525)
2626from zarr .core .dtype .common import HasEndianness
2727from zarr .core .dtype .wrapper import TBaseDType , TBaseScalar , ZDType
28- from zarr .core .group import Group , GroupMetadata
28+ from zarr .core .group import GroupMetadata
2929from zarr .core .metadata .v2 import ArrayV2Metadata
3030from zarr .core .metadata .v3 import ArrayV3Metadata
3131from zarr .core .sync import sync
3232from zarr .registry import get_codec_class
3333from zarr .storage import StorePath
3434
35- logger = logging .getLogger (__name__ )
35+ _logger = logging .getLogger (__name__ )
3636
3737
3838def migrate_v2_to_v3 (
@@ -41,10 +41,10 @@ def migrate_v2_to_v3(
4141 output_store : Store | None = None ,
4242 dry_run : bool = False ,
4343) -> None :
44- """Migrate all v2 metadata in a zarr hierarchy to v3.
44+ """Migrate all v2 metadata in a Zarr store to v3.
4545
46- This will create a zarr.json file at each level (for every group / array). v2 files (.zarray, .zattrs etc.)
47- will be left as-is.
46+ This will create a zarr.json file at each level of a Zarr hierarchy (for every group / array).
47+ v2 files (.zarray, .zattrs etc.) will be left as-is.
4848
4949 Parameters
5050 ----------
@@ -69,7 +69,7 @@ def migrate_v2_to_v3(
6969
7070
7171def migrate_to_v3 (zarr_v2 : Array | Group , output_path : StorePath , dry_run : bool = False ) -> None :
72- """Migrate all v2 metadata in a zarr array/group to v3.
72+ """Migrate all v2 metadata in a Zarr array/group to v3.
7373
7474 Note - if a group is provided, then all arrays / groups within this group will also be converted.
7575 A zarr.json file will be created for each level and written to output_path, with any v2 files
@@ -141,7 +141,7 @@ async def remove_metadata(
141141 if force or await _metadata_exists (
142142 cast (Literal [2 , 3 ], alternative_metadata ), store_path / parent_path
143143 ):
144- logger .info ("Deleting metadata at %s" , store_path / file_path )
144+ _logger .info ("Deleting metadata at %s" , store_path / file_path )
145145 if not dry_run :
146146 awaitables .append ((store_path / file_path ).delete ())
147147 else :
@@ -287,7 +287,7 @@ async def _save_v3_metadata(
287287 if await zarr_json_path .exists ():
288288 raise ValueError (f"{ ZARR_JSON } already exists at { zarr_json_path } " )
289289
290- logger .info ("Saving metadata to %s" , zarr_json_path )
290+ _logger .info ("Saving metadata to %s" , zarr_json_path )
291291 to_save = metadata_v3 .to_buffer_dict (default_buffer_prototype ())
292292
293293 if not dry_run :
0 commit comments