Skip to content

Commit b0bb86e

Browse files
authored
Remove unnecessary for loop when using get_axis_num (#8356)
1 parent 126b92a commit b0bb86e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

xarray/core/variable.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2162,7 +2162,7 @@ def rolling_window(
21622162
raise ValueError(
21632163
f"Expected {name}={arg!r} to be a scalar like 'dim'."
21642164
)
2165-
dim = [dim]
2165+
dim = (dim,)
21662166

21672167
# dim is now a list
21682168
nroll = len(dim)
@@ -2193,7 +2193,7 @@ def rolling_window(
21932193
pads[d] = (win - 1, 0)
21942194

21952195
padded = var.pad(pads, mode="constant", constant_values=fill_value)
2196-
axis = tuple(self.get_axis_num(d) for d in dim)
2196+
axis = self.get_axis_num(dim)
21972197
new_dims = self.dims + tuple(window_dim)
21982198
return Variable(
21992199
new_dims,

0 commit comments

Comments
 (0)