Fix for Zarr v3

cubed-dev · Aug 27, 2024 · 6a56586 · 6a56586
1 parent c090fe8
commit 6a56586
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/cubed/tests/primitive/test_blockwise.py b/cubed/tests/primitive/test_blockwise.py
@@ -1,6 +1,5 @@
 import numpy as np
 import pytest
-import zarr
 from numpy.testing import assert_array_equal
 
 from cubed.backend_array_api import namespace as nxp
@@ -279,10 +278,10 @@ def block_function(out_key):
 
     input = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
 
-    res1 = zarr.open_array(target_store1)
+    res1 = open_backend_array(target_store1, mode="r")
     assert_array_equal(res1[:], np.sqrt(input))
 
-    res2 = zarr.open_array(target_store2)
+    res2 = open_backend_array(target_store2, mode="r")
     assert_array_equal(res2[:], -np.sqrt(input))