check for rename method before lower_names

NREL · Nov 12, 2024 · 978f3a5 · 978f3a5
1 parent 8ae86d5
commit 978f3a5
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 12 deletions.
diff --git a/sup3r/preprocessing/accessor.py b/sup3r/preprocessing/accessor.py
@@ -90,7 +90,7 @@ def __init__(self, ds: Union[xr.Dataset, Self]):
         ds : xr.Dataset | xr.DataArray
             xarray Dataset instance to access with the following methods
         """
-        self._ds = lower_names(ds)
+        self._ds = lower_names(ds) if hasattr(ds, 'rename') else ds
         self._features = None
         self._meta = None
         self.time_slice = None

diff --git a/sup3r/preprocessing/utilities.py b/sup3r/preprocessing/utilities.py
@@ -23,17 +23,16 @@
 
 def lower_names(data):
     """Set all fields / coords / dims to lower case."""
-    return data.rename(
-        {
-            f: f.lower()
-            for f in [
-                *list(data.data_vars),
-                *list(data.dims),
-                *list(data.coords),
-            ]
-            if f != f.lower()
-        }
-    )
+    rename_map = {
+        f: f.lower()
+        for f in [
+            *list(data.data_vars),
+            *list(data.dims),
+            *list(data.coords),
+        ]
+        if f != f.lower()
+    }
+    return data.rename(rename_map)
 
 
 def get_input_handler_class(input_handler_name: Optional[str] = None):