pyright

Author: cmp0xff

pandas-stubs/core/indexes/range.pyi

@@ -4,7 +4,6 @@ from collections.abc import (
 )
 from typing import (
     Any,
-    final,
     overload,
 )
 
@@ -62,16 +61,16 @@ class RangeIndex(_IndexSubclassBase[int, np.int64]):
     ) -> tuple[np_1darray[np.intp], RangeIndex]: ...
     @property
     def size(self) -> int: ...
-    # Base class returns `Self`, but for `RangeIndex` that's not true.
-    def __floordiv__(  # type: ignore[override]
-        self, other: float | Sequence[float] | Index[int] | Index[float]
-    ) -> Index[int]: ...
     def all(self, *args: Any, **kwargs: Any) -> bool: ...
     def any(self, *args: Any, **kwargs: Any) -> bool: ...
-    @final
+    @overload
+    def union(  # type: ignore[override]
+        self, other: Sequence[int] | Index[int] | Self, sort: bool | None = None
+    ) -> Index[int] | Self: ...
+    @overload
     def union(  # type: ignore[override]
-        self, other: list[HashableT] | Index, sort: bool | None = None
-    ) -> Index | Index[int] | RangeIndex: ...
+        self, other: Sequence[HashableT] | Index, sort: bool | None = None
+    ) -> Index: ...
     @overload  # type: ignore[override]
     # pyrefly: ignore  # bad-override
     def __getitem__(

tests/indexes/test_indexes.py

@@ -5,7 +5,6 @@
 from typing import (
     TYPE_CHECKING,
     Any,
-    Union,
     cast,
 )
 
@@ -268,23 +267,21 @@ def test_range_index_union() -> None:
     check(
         assert_type(
             pd.RangeIndex(0, 10).union(pd.RangeIndex(10, 20)),
-            Union[pd.Index, "pd.Index[int]", pd.RangeIndex],
+            "pd.Index[int] | pd.RangeIndex",
         ),
         pd.RangeIndex,
     )
     check(
         assert_type(
-            pd.RangeIndex(0, 10).union([11, 12, 13]),
-            Union[pd.Index, "pd.Index[int]", pd.RangeIndex],
+            pd.RangeIndex(0, 10).union([11, 12, 13]), "pd.Index[int] | pd.RangeIndex"
         ),
         pd.Index,
+        int,
     )
     check(
-        assert_type(
-            pd.RangeIndex(0, 10).union(["a", "b", "c"]),
-            Union[pd.Index, "pd.Index[int]", pd.RangeIndex],
-        ),
+        assert_type(pd.RangeIndex(0, 10).union(["a", "b", "c"]), pd.Index),
         pd.Index,
+        str,
     )