[util] Cherry pick 37ac94f (master -> ndv2_dev)

pyxu-org · Mar 27, 2024 · c42d77d · c42d77d
1 parent 03b5f51
commit c42d77d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/pyxu/util/io.py b/src/pyxu/util/io.py
@@ -5,6 +5,7 @@
 
 import pyxu.info.deps as pxd
 import pyxu.info.ptype as pxt
+import pyxu.util.array_module as pxam
 
 __all__ = [
     "save_zarr",
@@ -35,7 +36,7 @@ def save_zarr(filedir: pxt.Path, kw_in: dict[str, pxt.NDArray]) -> None:
                         compute=True,
                     )
                 else:
-                    zarr.save(filedir / filename, array)
+                    zarr.save(filedir / filename, pxam.to_NUMPY(array))
         except Exception as e:
             print(f"Failed to save {filename}: {e}")