Handle non dask arrays (#65)

xpublish-community · Mar 25, 2024 · ba22a86 · ba22a86
1 parent cbd287a
commit ba22a86
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/xpublish_wms/grid.py b/xpublish_wms/grid.py
@@ -226,15 +226,22 @@ def project(self, da: xr.DataArray, crs: str) -> xr.DataArray:
             )
             y_chunks = da.cf["latitude"].chunks if da.cf["latitude"].chunks else y.shape
 
+            if da.chunks:
+                x_coord_array = dask_array.from_array(x, chunks=x_chunks)
+                y_coord_array = dask_array.from_array(y, chunks=y_chunks)
+            else:
+                x_coord_array = x
+                y_coord_array = y
+
             da = da.assign_coords(
                 {
                     "x": (
                         da.cf["longitude"].dims,
-                        dask_array.from_array(x, chunks=x_chunks),
+                        x_coord_array,
                     ),
                     "y": (
                         da.cf["latitude"].dims,
-                        dask_array.from_array(y, chunks=y_chunks),
+                        y_coord_array,
                     ),
                 },
             )