From 40c898c40c25a05f144367b28b1d16109d77336a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Torsten=20W=C3=B6rtwein?= Date: Sun, 24 Dec 2023 11:21:22 -0500 Subject: [PATCH] OffsetSeries inherits from Series[BaseOffset] --- pandas-stubs/_typing.pyi | 4 +++- pandas-stubs/core/series.pyi | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/pandas-stubs/_typing.pyi b/pandas-stubs/_typing.pyi index 5ab88069e..7210afae2 100644 --- a/pandas-stubs/_typing.pyi +++ b/pandas-stubs/_typing.pyi @@ -32,6 +32,7 @@ from typing_extensions import TypeAlias from pandas._libs.interval import Interval from pandas._libs.tslibs import ( + BaseOffset, Period, Timedelta, Timestamp, @@ -527,7 +528,8 @@ S1 = TypeVar( | datetime.timedelta # includes pd.Timedelta | Period | Interval - | CategoricalDtype, + | CategoricalDtype + | BaseOffset, ) IndexingInt: TypeAlias = ( diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi index 2a42e8a3d..a4fdeeb5d 100644 --- a/pandas-stubs/core/series.pyi +++ b/pandas-stubs/core/series.pyi @@ -2127,7 +2127,7 @@ class PeriodSeries(Series[Period]): def dt(self) -> PeriodProperties: ... # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride] def __sub__(self, other: PeriodSeries) -> OffsetSeries: ... # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride] -class OffsetSeries(Series): +class OffsetSeries(Series[BaseOffset]): @overload # type: ignore[override] def __radd__(self, other: Period) -> PeriodSeries: ... @overload