Remove unnecessary for loop when using get_axis_num (#8356)

pydata · Oct 22, 2023 · b0bb86e · b0bb86e
1 parent 126b92a
commit b0bb86e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/xarray/core/variable.py b/xarray/core/variable.py
@@ -2162,7 +2162,7 @@ def rolling_window(
                     raise ValueError(
                         f"Expected {name}={arg!r} to be a scalar like 'dim'."
                     )
-            dim = [dim]
+            dim = (dim,)
 
         # dim is now a list
         nroll = len(dim)
@@ -2193,7 +2193,7 @@ def rolling_window(
                 pads[d] = (win - 1, 0)
 
         padded = var.pad(pads, mode="constant", constant_values=fill_value)
-        axis = tuple(self.get_axis_num(d) for d in dim)
+        axis = self.get_axis_num(dim)
         new_dims = self.dims + tuple(window_dim)
         return Variable(
             new_dims,