bug with setitem with oindex and sharding #2834
Description
Zarr version
v3.0.2
Numcodecs version
?
Python Version
3.12
Operating System
any
Installation
any
Description
Discovered in #2825
Steps to reproduce
import zarr
import numpy as np
store = zarr.storage.MemoryStore()
group = zarr.group(store)
array = group.create_array(
name="zoo",
shape=(1,2,1),
chunks=(1,2,1),
shards=(1,2,1),
dtype=np.int32,
)
zindexer = (np.array([0]), np.array([0, 0]), np.array([0]))
new_data = np.full(array.oindex[zindexer].shape, fill_value=1)
array.oindex[zindexer] = new_dataraises the error:
ValueError: shape mismatch: value array of shape (1,2,1) could not be broadcast to indexing result of shape (2,)
It succeeds with shards=None.
Additional output
No response