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