Fix zarr-developers#1874

dcherian · May 14, 2024 · 3319359 · 3319359
1 parent 36c6253
commit 3319359
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/zarr/core.py b/zarr/core.py
@@ -2283,8 +2283,11 @@ def _process_for_setitem(self, ckey, chunk_selection, value, fields=None):
                 chunk.fill(value)
 
             else:
-                # ensure array is contiguous
-                chunk = value.astype(self._dtype, order=self._order, copy=False)
+                if not isinstance(value, np.ndarray):
+                    chunk = np.asarray(value, dtype=self._dtype, order=self._order)
+                else:
+                    # ensure array is contiguous
+                    chunk = value.astype(self._dtype, order=self._order, copy=False)
 
         else:
             # partially replace the contents of this chunk

diff --git a/zarr/tests/test_core.py b/zarr/tests/test_core.py
@@ -3167,3 +3167,6 @@ def test_scalar_indexing():
 
     assert store["a"][1] == np.array(2.0)
     assert store["a"][(1,)] == np.array(2.0)
+
+    store["a"][0] = [-1]
+    assert store["a"][0] == np.array(-1)