Replaces cf.axes with get_dim_keys

xCDAT · Mar 8, 2024 · e2125f9 · e2125f9
1 parent a5bf57c
commit e2125f9
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/xcdat/regridder/regrid2.py b/xcdat/regridder/regrid2.py
@@ -3,6 +3,7 @@
 import numpy as np
 import xarray as xr
 
+from xcdat.axis import get_dim_keys
 from xcdat.regridder.base import BaseRegridder, _preserve_bounds
 
 
@@ -479,10 +480,7 @@ def _pertub(value: np.ndarray) -> np.ndarray:
 
 
 def _get_dimension(input_data_var, cf_axis_name):
-    name = input_data_var.cf.axes[cf_axis_name]
-
-    if isinstance(name, list):
-        name = name[0]
+    name = get_dim_keys(input_data_var, cf_axis_name)
 
     index = input_data_var.dims.index(name)