Add MaskType to other cases

pandas-dev · Dec 13, 2024 · e299ca4 · e299ca4
1 parent b6ae6e8
commit e299ca4
Show file tree

Hide file tree

Showing 5 changed files with 11 additions and 9 deletions.
diff --git a/pandas-stubs/core/indexes/base.pyi b/pandas-stubs/core/indexes/base.pyi
@@ -48,6 +48,7 @@ from pandas._typing import (
     HashableT,
     Label,
     Level,
+    MaskType,
     NaPosition,
     TimedeltaDtypeArg,
     TimestampDtypeArg,
@@ -373,7 +374,7 @@ class Index(IndexOpsMixin[S1]):
             | np_ndarray_anyint
             | Sequence[int]
             | Index
-            | Series[bool]
+            | MaskType
             | Sequence[bool]
             | np_ndarray_bool
         ),

diff --git a/pandas-stubs/core/indexes/interval.pyi b/pandas-stubs/core/indexes/interval.pyi
@@ -13,7 +13,6 @@ import pandas as pd
 from pandas import Index
 from pandas.core.indexes.extension import ExtensionIndex
 from pandas.core.series import (
-    Series,
     TimedeltaSeries,
     TimestampSeries,
 )
@@ -31,6 +30,7 @@ from pandas._typing import (
     IntervalClosedType,
     IntervalT,
     Label,
+    MaskType,
     np_ndarray_anyint,
     np_ndarray_bool,
     npt,
@@ -254,7 +254,7 @@ class IntervalIndex(ExtensionIndex[IntervalT], IntervalMixin):
             | np_ndarray_anyint
             | Sequence[int]
             | Index
-            | Series[bool]
+            | MaskType
             | np_ndarray_bool
         ),
     ) -> IntervalIndex[IntervalT]: ...

diff --git a/pandas-stubs/core/indexes/multi.pyi b/pandas-stubs/core/indexes/multi.pyi
@@ -18,6 +18,7 @@ from pandas._typing import (
     Dtype,
     DtypeArg,
     HashableT,
+    MaskType,
     np_ndarray_anyint,
     np_ndarray_bool,
 )
@@ -119,7 +120,7 @@ class MultiIndex(Index[Any]):
             | np_ndarray_anyint
             | Sequence[int]
             | Index
-            | pd.Series[bool]
+            | MaskType
             | Sequence[bool]
             | np_ndarray_bool
         ),

diff --git a/pandas-stubs/core/indexes/range.pyi b/pandas-stubs/core/indexes/range.pyi
@@ -2,11 +2,11 @@ from collections.abc import Sequence
 from typing import overload
 
 import numpy as np
-from pandas import Series
 from pandas.core.indexes.base import Index
 
 from pandas._typing import (
     HashableT,
+    MaskType,
     np_ndarray_anyint,
     np_ndarray_bool,
     npt,
@@ -78,7 +78,7 @@ class RangeIndex(Index[int]):
             | np_ndarray_anyint
             | Sequence[int]
             | Index
-            | Series[bool]
+            | MaskType
             | Sequence[bool]
             | np_ndarray_bool
         ),

diff --git a/pandas-stubs/core/series.pyi b/pandas-stubs/core/series.pyi
@@ -1433,7 +1433,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             Series[S1]
             | Series[_bool]
             | np.ndarray
-            | Callable[[Series[S1]], Series[bool]]
+            | Callable[[Series[S1]], MaskType]
             | Callable[[S1], bool]
         ),
         other=...,
@@ -1448,7 +1448,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
             Series[S1]
             | Series[_bool]
             | np.ndarray
-            | Callable[[Series[S1]], Series[bool]]
+            | Callable[[Series[S1]], MaskType]
             | Callable[[S1], bool]
         ),
         other: Scalar | Series[S1] | DataFrame | Callable | NAType | None = ...,
@@ -1462,7 +1462,7 @@ class Series(IndexOpsMixin[S1], NDFrame):
         caselist: list[
             tuple[
                 Sequence[bool]
-                | Series[bool]
+                | MaskType
                 | Callable[[Series], Series | np.ndarray | Sequence[bool]],
                 ListLikeU | Scalar | Callable[[Series], Series | np.ndarray],
             ],