5
5
import numcodecs .abc
6
6
7
7
import zarr
8
+ from zarr import Array , Group
8
9
from zarr .abc .codec import ArrayArrayCodec , BytesBytesCodec , Codec
9
10
from zarr .abc .store import Store
10
11
from zarr .codecs .blosc import BloscCodec , BloscShuffle
11
12
from zarr .codecs .bytes import BytesCodec
12
13
from zarr .codecs .gzip import GzipCodec
13
14
from zarr .codecs .transpose import TransposeCodec
14
15
from zarr .codecs .zstd import ZstdCodec
15
- from zarr .core .array import Array
16
16
from zarr .core .buffer .core import default_buffer_prototype
17
17
from zarr .core .chunk_key_encodings import V2ChunkKeyEncoding
18
18
from zarr .core .common import (
25
25
)
26
26
from zarr .core .dtype .common import HasEndianness
27
27
from zarr .core .dtype .wrapper import TBaseDType , TBaseScalar , ZDType
28
- from zarr .core .group import Group , GroupMetadata
28
+ from zarr .core .group import GroupMetadata
29
29
from zarr .core .metadata .v2 import ArrayV2Metadata
30
30
from zarr .core .metadata .v3 import ArrayV3Metadata
31
31
from zarr .core .sync import sync
32
32
from zarr .registry import get_codec_class
33
33
from zarr .storage import StorePath
34
34
35
- logger = logging .getLogger (__name__ )
35
+ _logger = logging .getLogger (__name__ )
36
36
37
37
38
38
def migrate_v2_to_v3 (
@@ -41,10 +41,10 @@ def migrate_v2_to_v3(
41
41
output_store : Store | None = None ,
42
42
dry_run : bool = False ,
43
43
) -> None :
44
- """Migrate all v2 metadata in a zarr hierarchy to v3.
44
+ """Migrate all v2 metadata in a Zarr store to v3.
45
45
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.
48
48
49
49
Parameters
50
50
----------
@@ -69,7 +69,7 @@ def migrate_v2_to_v3(
69
69
70
70
71
71
def 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.
73
73
74
74
Note - if a group is provided, then all arrays / groups within this group will also be converted.
75
75
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(
141
141
if force or await _metadata_exists (
142
142
cast (Literal [2 , 3 ], alternative_metadata ), store_path / parent_path
143
143
):
144
- logger .info ("Deleting metadata at %s" , store_path / file_path )
144
+ _logger .info ("Deleting metadata at %s" , store_path / file_path )
145
145
if not dry_run :
146
146
awaitables .append ((store_path / file_path ).delete ())
147
147
else :
@@ -287,7 +287,7 @@ async def _save_v3_metadata(
287
287
if await zarr_json_path .exists ():
288
288
raise ValueError (f"{ ZARR_JSON } already exists at { zarr_json_path } " )
289
289
290
- logger .info ("Saving metadata to %s" , zarr_json_path )
290
+ _logger .info ("Saving metadata to %s" , zarr_json_path )
291
291
to_save = metadata_v3 .to_buffer_dict (default_buffer_prototype ())
292
292
293
293
if not dry_run :
0 commit comments