Fix test_update_ds_from_layer_ds

gwmod · Sep 19, 2024 · 555e0f0 · 555e0f0
1 parent b1810e8
commit 555e0f0
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/nlmod/read/regis.py b/nlmod/read/regis.py
@@ -135,7 +135,11 @@ def get_regis(
     regis_ds : xarray dataset
         dataset with regis data projected on the modelgrid.
     """
-    ds = xr.open_dataset(REGIS_URL, decode_times=False)
+    ds = xr.open_dataset(REGIS_URL, decode_coords="all")
+    if "crs" in ds.coords:
+        # remove the crs coordinate, as rioxarray does not recognise the crs
+        # and we set the crs at the end of this method by hand
+        ds = ds.drop_vars("crs")
 
     # set x and y dimensions to cell center
     ds["x"] = ds.x_bounds.mean("bounds")
@@ -187,7 +191,6 @@ def get_regis(
     ds.attrs["gridtype"] = "structured"
     ds.attrs["extent"] = extent
     for datavar in ds:
-        ds[datavar].attrs["grid_mapping"] = "crs"
         ds[datavar].attrs["source"] = "REGIS"
         ds[datavar].attrs["url"] = REGIS_URL
         ds[datavar].attrs["date"] = dt.datetime.now().strftime("%Y%m%d")

diff --git a/tests/test_026_grid.py b/tests/test_026_grid.py
@@ -29,7 +29,7 @@ def get_regis():
             os.makedirs(model_ws)
         regis = nlmod.read.regis.get_regis(extent)
         regis.to_netcdf(fname)
-    return xr.open_dataset(fname)
+    return xr.open_dataset(fname, decode_coords="all")
 
 
 def get_structured_model_ds():