Skip to content

Commit 54b2ef1

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 13d6bdf commit 54b2ef1

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

xarray/core/toolzcompat.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -51,5 +51,6 @@ def sliding_window(n, seq):
5151
*(
5252
collections.deque(itertools.islice(it, i), 0) or it
5353
for i, it in enumerate(itertools.tee(seq, n))
54-
)
54+
),
55+
strict=False,
5556
)

xarray/groupers.py

+8-6
Original file line numberDiff line numberDiff line change
@@ -24,14 +24,14 @@
2424
from xarray.core.groupby import T_Group, _DummyGroup
2525
from xarray.core.indexes import safe_cast_to_index
2626
from xarray.core.resample_cftime import CFTimeGrouper
27+
from xarray.core.toolzcompat import sliding_window
2728
from xarray.core.types import (
2829
Bins,
2930
DatetimeLike,
3031
GroupIndices,
3132
ResampleCompatible,
3233
SideOptions,
3334
)
34-
from xarray.core.toolzcompat import sliding_window
3535
from xarray.core.variable import Variable
3636

3737
__all__ = [
@@ -495,7 +495,7 @@ def season_to_month_tuple(seasons: Sequence[str]) -> tuple[tuple[int, ...], ...]
495495
initials = "JFMAMJJASOND"
496496
starts = dict(
497497
("".join(s), i + 1)
498-
for s, i in zip(sliding_window(2, initials + "J"), range(12))
498+
for s, i in zip(sliding_window(2, initials + "J"), range(12), strict=False)
499499
)
500500
result: list[tuple[int, ...]] = []
501501
for i, season in enumerate(seasons):
@@ -602,7 +602,7 @@ class SeasonResampler(Resampler):
602602

603603
def __post_init__(self):
604604
self.season_inds = season_to_month_tuple(self.seasons)
605-
self.season_tuples = dict(zip(self.seasons, self.season_inds))
605+
self.season_tuples = dict(zip(self.seasons, self.season_inds, strict=False))
606606

607607
def factorize(self, group):
608608
if group.ndim != 1:
@@ -624,7 +624,7 @@ def factorize(self, group):
624624
season_label = np.full(group.shape, "", dtype=f"U{nstr}")
625625

626626
# offset years for seasons with December and January
627-
for season_str, season_ind in zip(seasons, season_inds):
627+
for season_str, season_ind in zip(seasons, season_inds, strict=False):
628628
season_label[month.isin(season_ind)] = season_str
629629
if "DJ" in season_str:
630630
after_dec = season_ind[season_str.index("D") + 1 :]
@@ -651,14 +651,16 @@ def factorize(self, group):
651651
)
652652

653653
sbins = first_items.values.astype(int)
654-
group_indices = [slice(i, j) for i, j in zip(sbins[:-1], sbins[1:])]
654+
group_indices = [
655+
slice(i, j) for i, j in zip(sbins[:-1], sbins[1:], strict=False)
656+
]
655657
group_indices += [slice(sbins[-1], None)]
656658

657659
# Make sure the first and last timestamps
658660
# are for the correct months,if not we have incomplete seasons
659661
unique_codes = np.arange(len(unique_coord))
660662
if self.drop_incomplete:
661-
for idx, slicer in zip([0, -1], (slice(1, None), slice(-1))):
663+
for idx, slicer in zip([0, -1], (slice(1, None), slice(-1)), strict=False):
662664
stamp_year, stamp_season = frame.index[idx]
663665
code = seasons.index(stamp_season)
664666
stamp_month = season_inds[code][idx]

0 commit comments

Comments
 (0)