Setting lon and lat as coordinates

Author: erikvansebille

parcels/xgrid.py

@@ -101,10 +101,12 @@ def __init__(self, grid: xgcm.Grid, mesh="flat"):
         self._mesh = mesh
         self._spatialhash = None
         ds = grid._ds
-        if "lon" in ds and hasattr(ds["lon"], "load"):
-            ds["lon"].load()
-        if "lat" in ds and hasattr(ds["lat"], "load"):
-            ds["lat"].load()
+
+        # Set the coordinates for the dataset (needed to be done explicitly for curvilinear grids)
+        if "lon" in ds:
+            ds.set_coords("lon")
+        if "lat" in ds:
+            ds.set_coords("lat")
 
         if len(set(grid.axes) & {"X", "Y", "Z"}) > 0:  # Only if spatial grid is >0D (see #2054 for further development)
             assert_valid_lat_lon(ds["lat"], ds["lon"], grid.axes)