merge

Author: yangdanny97

pandas-stubs/_typing.pyi

@@ -847,6 +847,26 @@ S1 = TypeVar("S1", bound=SeriesDType, default=Any)
 S2 = TypeVar("S2", bound=SeriesDType)
 S3 = TypeVar("S3", bound=SeriesDType)
 
+# Constraint, instead of bound
+C2 = TypeVar(
+    "C2",
+    str,
+    bytes,
+    datetime.date,
+    datetime.time,
+    bool,
+    int,
+    float,
+    complex,
+    Dtype,
+    datetime.datetime,  # includes pd.Timestamp
+    datetime.timedelta,  # includes pd.Timedelta
+    Period,
+    Interval,
+    CategoricalDtype,
+    BaseOffset,
+)
+
 IndexingInt: TypeAlias = (
     int | np.int_ | np.integer | np.unsignedinteger | np.signedinteger | np.int8
 )

pandas-stubs/core/indexes/base.pyi

@@ -34,14 +34,15 @@ from pandas import (
 )
 from pandas.core.arrays import ExtensionArray
 from pandas.core.base import IndexOpsMixin
-from pandas.core.strings import StringMethods
+from pandas.core.strings.accessor import StringMethods
 from typing_extensions import (
     Never,
     Self,
 )
 
 from pandas._libs.interval import _OrderableT
 from pandas._typing import (
+    C2,
     S1,
     AnyAll,
     ArrayLike,
@@ -411,7 +412,12 @@ class Index(IndexOpsMixin[S1]):
     ) -> Self: ...
     @overload
     def __getitem__(self, idx: int | tuple[np_ndarray_anyint, ...]) -> S1: ...
-    def append(self, other): ...
+    @overload
+    def append(
+        self: Index[C2], other: Index[C2] | Sequence[Index[C2]]
+    ) -> Index[C2]: ...
+    @overload
+    def append(self, other: Index | Sequence[Index]) -> Index: ...
     def putmask(self, mask, value): ...
     def equals(self, other) -> bool: ...
     @final

pandas-stubs/core/indexes/multi.pyi

@@ -57,7 +57,7 @@ class MultiIndex(Index):
     @classmethod
     def from_product(
         cls,
-        iterables: Sequence[SequenceNotStr[Hashable] | pd.Series | pd.Index],
+        iterables: Sequence[SequenceNotStr[Hashable] | pd.Series | pd.Index | range],
         sortorder: int | None = ...,
         names: SequenceNotStr[Hashable] = ...,
     ) -> Self: ...
@@ -132,7 +132,7 @@ class MultiIndex(Index):
     def __getitem__(  # pyright: ignore[reportIncompatibleMethodOverride]
         self, key: int
     ) -> tuple: ...
-    def append(self, other): ...
+    def append(self, other): ...  # pyrefly: ignore
     def argsort(self, *args, **kwargs): ...
     def repeat(self, repeats, axis=...): ...
     def drop(self, codes, level: Level | None = None, errors: str = "raise") -> Self: ...  # type: ignore[override] # pyright: ignore[reportIncompatibleMethodOverride]