ENH: Improve typing for Interval

pandas-stubs/_libs/interval.pyi

@@ -1,18 +1,23 @@
 from typing import (
     Any,
     Generic,
+    Literal,
     TypeVar,
     overload,
 )
 
 import numpy as np
 from pandas import (
+    IntervalIndex,
+    Series,
     Timedelta,
     Timestamp,
 )
 
 from pandas._typing import (
     IntervalClosedType,
+    IntervalT,
+    np_ndarray_bool,
     npt,
 )
 
@@ -152,6 +157,44 @@ class Interval(IntervalMixin, Generic[_OrderableT]):
     def overlaps(self: Interval[int], other: Interval[float]) -> bool: ...
     @overload
     def overlaps(self: Interval[float], other: Interval[int]) -> bool: ...
+    @overload
+    def __gt__(self, other: Interval[_OrderableT]) -> bool: ...
+    @overload
+    def __gt__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __gt__(self, other: Series[_OrderableT]) -> Series[bool]: ...
+    @overload
+    def __lt__(self, other: Interval[_OrderableT]) -> bool: ...
+    @overload
+    def __lt__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __lt__(self, other: Series[_OrderableT]) -> Series[bool]: ...
+    @overload
+    def __ge__(self, other: Interval[_OrderableT]) -> bool: ...
+    @overload
+    def __ge__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __ge__(self, other: Series[_OrderableT]) -> Series[bool]: ...
+    @overload
+    def __le__(self, other: Interval[_OrderableT]) -> bool: ...
+    @overload
+    def __le__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __eq__(self, other: Interval[_OrderableT]) -> bool: ...  # type: ignore[misc]
+    @overload
+    def __eq__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __eq__(self, other: Series[_OrderableT]) -> Series[bool]: ...  # type: ignore[misc]
+    @overload
+    def __eq__(self, other: object) -> Literal[False]: ...
+    @overload
+    def __ne__(self, other: Interval[_OrderableT]) -> bool: ...  # type: ignore[misc]
+    @overload
+    def __ne__(self: IntervalT, other: IntervalIndex[IntervalT]) -> np_ndarray_bool: ...  # type: ignore[misc]
+    @overload
+    def __ne__(self, other: Series[_OrderableT]) -> Series[bool]: ...  # type: ignore[misc]
+    @overload
+    def __ne__(self, other: object) -> Literal[True]: ...
 
 class IntervalTree(IntervalMixin):
     def __init__(

pandas-stubs/_typing.pyi

@@ -197,7 +197,10 @@ S1 = TypeVar(
     Timedelta,
     np.datetime64,
     Period,
-    Interval,
+    Interval[int],
+    Interval[float],
+    Interval[Timestamp],
+    Interval[Timedelta],
 )
 T1 = TypeVar(
     "T1", str, int, np.int64, np.uint64, np.float64, float, np.dtype[np.generic]
@@ -222,7 +225,13 @@ NDFrameT = TypeVar("NDFrameT", bound=NDFrame)
 IndexT = TypeVar("IndexT", bound=Index)
 
 # Interval closed type
-
+IntervalT = TypeVar(
+    "IntervalT",
+    Interval[int],
+    Interval[float],
+    Interval[Timestamp],
+    Interval[Timedelta],
+)
 IntervalClosedType: TypeAlias = Literal["left", "right", "both", "neither"]
 
 IgnoreRaiseCoerce: TypeAlias = Literal["ignore", "raise", "coerce"]

pandas-stubs/core/algorithms.pyi

@@ -1,6 +1,5 @@
 from typing import (
     Sequence,
-    TypeVar,
     overload,
 )
 
@@ -9,32 +8,26 @@ from pandas import (
     Categorical,
     CategoricalIndex,
     Index,
-    Interval,
     IntervalIndex,
     PeriodIndex,
     Series,
 )
 from pandas.api.extensions import ExtensionArray
 
-from pandas._typing import AnyArrayLike
+from pandas._typing import (
+    AnyArrayLike,
+    IntervalT,
+)
 
 # These are type: ignored because the Index types overlap due to inheritance but indices
 # with extension types return the same type while standard type return ndarray
 
-_IntervalT = TypeVar(
-    "_IntervalT",
-    Interval[int],
-    Interval[float],
-    Interval[pd.Timestamp],
-    Interval[pd.Timedelta],
-)
-
 @overload
 def unique(values: PeriodIndex) -> PeriodIndex: ...  # type: ignore[misc]
 @overload
 def unique(values: CategoricalIndex) -> CategoricalIndex: ...  # type: ignore[misc]
 @overload
-def unique(values: IntervalIndex[_IntervalT]) -> IntervalIndex[_IntervalT]: ...  # type: ignore[misc]
+def unique(values: IntervalIndex[IntervalT]) -> IntervalIndex[IntervalT]: ...  # type: ignore[misc]
 @overload
 def unique(values: Index) -> np.ndarray: ...
 @overload