Fixing bug in DatasetZarr when backend is not set

Signed-off-by: Julio Faracco <[email protected]>
discovery-unicamp · Oct 1, 2023 · f952374 · f952374
1 parent c7dc72f
commit f952374
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/dasf/datasets/base.py b/dasf/datasets/base.py
@@ -503,9 +503,9 @@ def _lazy_load(self, xp, **kwargs):
             store[".zarray"] = json.dumps(meta).encode()
 
             array = zarr.open_array(store, meta_array=xp.empty(()))
-            return da.from_zarr(array, chunks=self._chunks).map_blocks(xp.asarray)
+            return da.from_zarr(array, chunks=array.chunks).map_blocks(xp.asarray)
 
-        return da.from_zarr(self._root_file, chunks=array.chunks).map_blocks(xp.asarray)
+        return da.from_zarr(self._root_file, chunks=self._chunks).map_blocks(xp.asarray)
 
     def _load(self, xp, **kwargs):
         """Load data using CPU container.