fixing for regions i.e. land_60N60S

E3SM-Project · Nov 8, 2024 · 542b88b · 542b88b
1 parent ef261b8
commit 542b88b
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/e3sm_diags/driver/utils/regrid.py b/e3sm_diags/driver/utils/regrid.py
@@ -479,7 +479,9 @@ def _get_region_mask_var_key(ds_mask: xr.Dataset, region: str):
     ValueError
         If the region passed is not land or ocean.
     """
-    region_keys = FRAC_REGION_KEYS.get(region)
+    for region_prefix in ["land", "ocean"]:
+        if region_prefix in region:
+            region_keys = FRAC_REGION_KEYS.get(region_prefix)
 
     if region_keys is None:
         raise ValueError(f"Only land and ocean regions are supported, not '{region}'.")