TypeError: ArrayV2Metadata.__init__() got an unexpected keyword argument 'chunk_grid' when updating attrs on a v2 array #2269
Description
Zarr version
v3
Numcodecs version
na
Python Version
na
Operating System
na
Installation
na
Description
data_type is also affected. IIUC, these are called chunks and dtype in v2. We need to mark them as dataclasses.field(init=False) or somehow handle renaming them on the way in.
Steps to reproduce
import numpy as np
from zarr.storage import MemoryStore, StorePath
from zarr import Array
store = StorePath(MemoryStore(mode="w"), path="/")
data = np.arange(0, 8, dtype="uint16")
a = Array.create(
store / "simple_v2",
zarr_format=2,
shape=data.shape,
chunks=(4,),
dtype=data.dtype,
fill_value=0,
)
a.attrs.put({"key": 0})raises with
---------------------------------------------------------------------------
TypeError Traceback (most recent call last)
Cell In[10], line 10
1 data = np.arange(0, 8, dtype="uint16")
2 a = Array.create(
3 store,
4 zarr_format=2,
(...)
8 fill_value=0,
9 )
---> 10 a.attrs.put({"key": 0})
File ~/gh/zarr-developers/zarr-python/src/zarr/core/attributes.py:53, in Attributes.put(self, d)
39 def put(self, d: dict[str, JSON]) -> None:
40 """
41 Overwrite all attributes with the values from `d`.
42
(...)
51 {'a': 3, 'c': 4}
52 """
---> 53 self._obj = self._obj.update_attributes(d)
File ~/gh/zarr-developers/zarr-python/src/zarr/core/array.py:2318, in Array.update_attributes(self, new_attributes)
2316 def update_attributes(self, new_attributes: dict[str, JSON]) -> Array:
2317 return type(self)(
-> 2318 sync(
2319 self._async_array.update_attributes(new_attributes),
2320 )
2321 )
File ~/gh/zarr-developers/zarr-python/src/zarr/core/sync.py:91, in sync(coro, loop, timeout)
88 return_result = next(iter(finished)).result()
90 if isinstance(return_result, BaseException):
---> 91 raise return_result
92 else:
93 return return_result
File ~/gh/zarr-developers/zarr-python/src/zarr/core/sync.py:50, in _runner(coro)
45 """
46 Await a coroutine and return the result of running it. If awaiting the coroutine raises an
47 exception, the exception will be returned.
48 """
49 try:
---> 50 return await coro
51 except Exception as ex:
52 return ex
File ~/gh/zarr-developers/zarr-python/src/zarr/core/array.py:753, in AsyncArray.update_attributes(self, new_attributes)
752 async def update_attributes(self, new_attributes: dict[str, JSON]) -> AsyncArray:
--> 753 new_metadata = self.metadata.update_attributes(new_attributes)
755 # Write new metadata
756 await self._save_metadata(new_metadata)
File ~/gh/zarr-developers/zarr-python/src/zarr/core/metadata/v2.py:174, in ArrayV2Metadata.update_attributes(self, attributes)
173 def update_attributes(self, attributes: dict[str, JSON]) -> Self:
--> 174 return replace(self, attributes=attributes)
File ~/mambaforge/envs/python=3.12/lib/python3.12/dataclasses.py:1588, in replace(obj, **changes)
1581 changes[f.name] = getattr(obj, f.name)
1583 # Create the new object, which calls __init__() and
1584 # __post_init__() (if defined), using all of the init fields we've
1585 # added and/or left in 'changes'. If there are values supplied in
1586 # changes that aren't fields, this will correctly raise a
1587 # TypeError.
-> 1588 return obj.__class__(**changes)
TypeError: ArrayV2Metadata.__init__() got an unexpected keyword argument 'chunk_grid'Additional output
No response