Skip to content

Commit 666da42

Browse files
Apply ruff rule RUF007
RUF007 Prefer `itertools.pairwise()` over `zip()` when iterating over successive pairs
1 parent 8aab23e commit 666da42

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

pyproject.toml

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -249,7 +249,6 @@ ignore = [
249249
"RUF002",
250250
"RUF003",
251251
"RUF005",
252-
"RUF007",
253252
"RUF012",
254253
]
255254
extend-select = [

xarray/groupers.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,7 @@
99
import datetime
1010
from abc import ABC, abstractmethod
1111
from dataclasses import dataclass, field
12+
from itertools import pairwise
1213
from typing import TYPE_CHECKING, Any, Literal, cast
1314

1415
import numpy as np
@@ -496,8 +497,7 @@ def factorize(self, group: T_Group) -> EncodedGroups:
496497
full_index, first_items, codes_ = self._get_index_and_items()
497498
sbins = first_items.values.astype(np.int64)
498499
group_indices: GroupIndices = tuple(
499-
[slice(i, j) for i, j in zip(sbins[:-1], sbins[1:], strict=True)]
500-
+ [slice(sbins[-1], None)]
500+
[slice(i, j) for i, j in pairwise(sbins)] + [slice(sbins[-1], None)]
501501
)
502502

503503
unique_coord = Variable(

xarray/tests/test_groupby.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@
33
import datetime
44
import operator
55
import warnings
6+
from itertools import pairwise
67
from unittest import mock
78

89
import numpy as np
@@ -1732,7 +1733,7 @@ def test_groupby_bins_multidim(self) -> None:
17321733
bincoord = np.array(
17331734
[
17341735
pd.Interval(left, right, closed="right")
1735-
for left, right in zip(bins[:-1], bins[1:], strict=True)
1736+
for left, right in pairwise(bins)
17361737
],
17371738
dtype=object,
17381739
)

0 commit comments

Comments
 (0)