Skip to content

Commit 5000dcb

Browse files
committed
rename base dtype, change type to kind
1 parent d74e7a4 commit 5000dcb

File tree

6 files changed

+130
-105
lines changed

6 files changed

+130
-105
lines changed

src/zarr/core/_info.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77

88
from zarr.abc.codec import ArrayArrayCodec, ArrayBytesCodec, BytesBytesCodec
99
from zarr.core.common import ZarrFormat
10-
from zarr.core.metadata.dtype import BaseDataType
10+
from zarr.core.metadata.dtype import DtypeBase
1111

1212
# from zarr.core.metadata.v3 import DataType
1313

@@ -80,7 +80,7 @@ class ArrayInfo:
8080

8181
_type: Literal["Array"] = "Array"
8282
_zarr_format: ZarrFormat
83-
_data_type: np.dtype[Any] | BaseDataType
83+
_data_type: np.dtype[Any] | DtypeBase
8484
_shape: tuple[int, ...]
8585
_shard_shape: tuple[int, ...] | None = None
8686
_chunk_shape: tuple[int, ...] | None = None

src/zarr/core/array.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@
9898
ArrayV3MetadataDict,
9999
T_ArrayMetadata,
100100
)
101-
from zarr.core.metadata.dtype import BaseDataType
101+
from zarr.core.metadata.dtype import DtypeBase
102102
from zarr.core.metadata.v2 import (
103103
_default_compressor,
104104
_default_filters,
@@ -679,7 +679,7 @@ def _create_metadata_v3(
679679
"""
680680

681681
shape = parse_shapelike(shape)
682-
codecs = list(codecs) if codecs is not None else _get_default_codecs(np.dtype(dtype))
682+
codecs = list(codecs) if codecs is not None else _get_default_codecs(dtype)
683683
chunk_key_encoding_parsed: ChunkKeyEncodingLike
684684
if chunk_key_encoding is None:
685685
chunk_key_encoding_parsed = {"name": "default", "separator": "/"}
@@ -1684,7 +1684,7 @@ async def info_complete(self) -> Any:
16841684
def _info(
16851685
self, count_chunks_initialized: int | None = None, count_bytes_stored: int | None = None
16861686
) -> Any:
1687-
_data_type: np.dtype[Any] | BaseDataType
1687+
_data_type: np.dtype[Any] | DtypeBase
16881688
if isinstance(self.metadata, ArrayV2Metadata):
16891689
_data_type = self.metadata.dtype
16901690
else:
@@ -4207,9 +4207,9 @@ def _get_default_chunk_encoding_v3(
42074207
"""
42084208
dtype = get_data_type_from_numpy(np_dtype)
42094209

4210-
default_filters = zarr_config.get("array.v3_default_filters").get(dtype.type)
4211-
default_serializer = zarr_config.get("array.v3_default_serializer").get(dtype.type)
4212-
default_compressors = zarr_config.get("array.v3_default_compressors").get(dtype.type)
4210+
default_filters = zarr_config.get("array.v3_default_filters").get(dtype.kind)
4211+
default_serializer = zarr_config.get("array.v3_default_serializer").get(dtype.kind)
4212+
default_compressors = zarr_config.get("array.v3_default_compressors").get(dtype.kind)
42134213

42144214
filters = tuple(_parse_array_array_codec(codec_dict) for codec_dict in default_filters)
42154215
serializer = _parse_array_bytes_codec(default_serializer)

src/zarr/core/common.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,6 @@
1919
import numpy as np
2020

2121
from zarr.core.config import config as zarr_config
22-
from zarr.core.strings import _VLEN_STRING_DTYPE
2322

2423
if TYPE_CHECKING:
2524
from collections.abc import Awaitable, Callable, Iterator
@@ -167,13 +166,7 @@ def parse_bool(data: Any) -> bool:
167166
raise ValueError(f"Expected bool, got {data} instead.")
168167

169168

170-
def parse_dtype(dtype: Any, zarr_format: ZarrFormat) -> np.dtype[Any]:
171-
if dtype is str or dtype == "str":
172-
if zarr_format == 2:
173-
# special case as object
174-
return np.dtype("object")
175-
else:
176-
return _VLEN_STRING_DTYPE
169+
def parse_dtype(dtype: Any) -> np.dtype[Any]:
177170
return np.dtype(dtype)
178171

179172

0 commit comments

Comments
 (0)