TypeError: ArrayV2Metadata.__init__() got an unexpected keyword argument 'chunk_grid' when updating attrs on a v2 array

[TomAugspurger]

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

[TomAugspurger]

This was closed by #2301.