Skip to content

Commit 9cf063f

Browse files
GH1419 Soften typing in RangeIndex.where
1 parent b1c7af9 commit 9cf063f

File tree

2 files changed

+3
-4
lines changed

2 files changed

+3
-4
lines changed

pandas-stubs/core/indexes/range.pyi

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,6 @@ from pandas.core.indexes.base import (
1515
Index,
1616
_IndexSubclassBase,
1717
)
18-
from typing_extensions import Self
1918

2019
from pandas._typing import (
2120
AnyArrayLike,
@@ -99,8 +98,8 @@ class RangeIndex(_IndexSubclassBase[int, np.int64]):
9998
def __getitem__( # pyright: ignore[reportIncompatibleMethodOverride]
10099
self, idx: int
101100
) -> int: ...
102-
def where(
101+
def where( # type: ignore[override]
103102
self,
104103
cond: Sequence[bool] | np_ndarray_bool | BooleanArray | IndexOpsMixin[bool],
105104
other: Scalar | AnyArrayLike | None = None,
106-
) -> Self: ...
105+
) -> Index: ...

tests/indexes/test_indexes.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1577,4 +1577,4 @@ def test_datetimeindex_where() -> None:
15771577
check(assert_type(val_scalar, pd.Index), pd.Index)
15781578

15791579
val_range = pd.RangeIndex(2).where(pd.Series([True, False]), 3)
1580-
check(assert_type(val_range, pd.RangeIndex), pd.RangeIndex)
1580+
check(assert_type(val_range, pd.Index), pd.RangeIndex)

0 commit comments

Comments
 (0)