OffsetSeries inherits from Series[BaseOffset] (#844)

pandas-dev · Dec 26, 2023 · 146cf23 · 146cf23
1 parent 72b22b5
commit 146cf23
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git 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
@@ -2137,7 +2137,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