Configuration of cache_netcdf usages

nlmod/cache.py

@@ -628,6 +628,9 @@ def ds_contains(ds, coords_2d=False, coords_3d=False, coords_time=False, datavar
         coords.append("x")
         coords.append("y")
         attrs.append("extent")
+
+        if "gridtype" in ds.attrs:
+            attrs.append("gridtype")
 
         if "angrot" in ds.attrs:
             attrs.append("angrot")

nlmod/dims/grid.py

@@ -1846,7 +1846,7 @@ def get_vertices(ds, vert_per_cid=4, epsilon=0, rotated=False):
     return vertices_da
 
 
-@cache.cache_netcdf()
+@cache.cache_netcdf(coords_2d=True)
 def mask_model_edge(ds, idomain=None):
     """get data array which is 1 for every active cell (defined by idomain) at
     the boundaries of the model (xmin, xmax, ymin, ymax). Other cells are 0.

nlmod/read/ahn.py

@@ -19,7 +19,7 @@
 logger = logging.getLogger(__name__)
 
 
-@cache.cache_netcdf()
+@cache.cache_netcdf(coords_2d=True)
 def get_ahn(ds=None, identifier="AHN4_DTM_5m", method="average", extent=None):
     """Get a model dataset with ahn variable.
 

nlmod/read/knmi.py

@@ -13,7 +13,7 @@
 logger = logging.getLogger(__name__)
 
 
-@cache.cache_netcdf()
+@cache.cache_netcdf(coords_2d=True, coords_time=True)
 def get_recharge(ds, method="linear", most_common_station=False):
     """add multiple recharge packages to the groundwater flow model with knmi
     data by following these steps:

nlmod/read/rws.py

@@ -37,7 +37,7 @@ def get_gdf_surface_water(ds):
     return gdf_swater
 
 
-@cache.cache_netcdf()
+@cache.cache_netcdf(coords_2d=True)
 def get_surface_water(ds, da_basename):
     """create 3 data-arrays from the shapefile with surface water: