allow rectifying dataset for nd

xcube-dev · Dec 20, 2024 · 234a3d8 · 234a3d8
1 parent fa79e87
commit 234a3d8
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 391 deletions.
diff --git a/xcube/core/gridmapping/regular.py b/xcube/core/gridmapping/regular.py
@@ -55,7 +55,9 @@ def _new_xy_coords(self) -> xr.DataArray:
         xy_coords = da.concatenate(
             [da.expand_dims(x_coords_2d, 0), da.expand_dims(y_coords_2d, 0)]
         )
-        xy_coords = da.rechunk(xy_coords, chunks=(2, 512, 512))
+        xy_coords = da.rechunk(
+            xy_coords, chunks=(2, xy_coords.chunksize[1], xy_coords.chunksize[2])
+        )
         xy_coords = xr.DataArray(
             xy_coords,
             dims=("coord", self.y_coords.dims[0], self.x_coords.dims[0]),