Replace Series[bool] and alike with MaskType

Author: loicdiridollou

pandas-stubs/core/indexes/base.pyi

@@ -48,6 +48,7 @@ from pandas._typing import (
     HashableT,
     Label,
     Level,
+    MaskType,
     NaPosition,
     TimedeltaDtypeArg,
     TimestampDtypeArg,
@@ -368,15 +369,7 @@ class Index(IndexOpsMixin[S1]):
     @overload
     def __getitem__(
         self,
-        idx: (
-            slice
-            | np_ndarray_anyint
-            | Sequence[int]
-            | Index
-            | Series[bool]
-            | Sequence[bool]
-            | np_ndarray_bool
-        ),
+        idx: slice | np_ndarray_anyint | Sequence[int] | Index | MaskType,
     ) -> Self: ...
     @overload
     def __getitem__(self, idx: int | tuple[np_ndarray_anyint, ...]) -> S1: ...

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]: ...

pandas-stubs/core/indexes/multi.pyi

@@ -18,6 +18,7 @@ from pandas._typing import (
     Dtype,
     DtypeArg,
     HashableT,
+    MaskType,
     np_ndarray_anyint,
     np_ndarray_bool,
 )
@@ -114,15 +115,7 @@ class MultiIndex(Index[Any]):
     @overload  # type: ignore[override]
     def __getitem__(
         self,
-        idx: (
-            slice
-            | np_ndarray_anyint
-            | Sequence[int]
-            | Index
-            | pd.Series[bool]
-            | Sequence[bool]
-            | np_ndarray_bool
-        ),
+        idx: slice | np_ndarray_anyint | Sequence[int] | Index | MaskType,
     ) -> Self: ...
     @overload
     def __getitem__(  # pyright: ignore[reportIncompatibleMethodOverride]

pandas-stubs/core/indexes/range.pyi

@@ -2,13 +2,12 @@ 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,
 )
 
@@ -73,15 +72,7 @@ class RangeIndex(Index[int]):
     @overload  # type: ignore[override]
     def __getitem__(
         self,
-        idx: (
-            slice
-            | np_ndarray_anyint
-            | Sequence[int]
-            | Index
-            | Series[bool]
-            | Sequence[bool]
-            | np_ndarray_bool
-        ),
+        idx: slice | np_ndarray_anyint | Sequence[int] | Index | MaskType,
     ) -> Index: ...
     @overload
     def __getitem__(  # pyright: ignore[reportIncompatibleMethodOverride]