Updates:

    - combine two str.split overloads and keep only _TS and _TS2;
    - fix test_indexes.py test for test_str_split().

Author: pan-vlados

pandas-stubs/core/strings.pyi

@@ -69,13 +69,9 @@ class StringMethods(NoNewAttributesMixin, Generic[T, _TS, _TM, _TS2]):
     ) -> _TS: ...
     @overload
     def split(
-        self, pat: str = ..., *, n: int = ..., expand: Literal[False], regex: bool = ...
+        self, pat: str = ..., *, n: int = ..., expand: Literal[False] = ..., regex: bool = ...
     ) -> _TS2: ...
     @overload
-    def split(
-        self, pat: str = ..., *, n: int = ..., expand: bool = ..., regex: bool = ...
-    ) -> T: ...
-    @overload
     def rsplit(self, pat: str = ..., *, n: int = ..., expand: Literal[True]) -> _TS: ...
     @overload
     def rsplit(self, pat: str = ..., *, n: int = ..., expand: bool = ...) -> T: ...

tests/test_indexes.py

@@ -111,7 +111,7 @@ def test_difference_none() -> None:
 def test_str_split() -> None:
     # GH 194
     ind = pd.Index(["a-b", "c-d"])
-    check(assert_type(ind.str.split("-"), "pd.Index[str]"), pd.Index)
+    check(assert_type(ind.str.split("-"), "pd.Index[list[str]]"), pd.Index)
     check(assert_type(ind.str.split("-", expand=True), pd.MultiIndex), pd.MultiIndex)
     check(
         assert_type(ind.str.split("-", expand=False), "pd.Index[list[str]]"), pd.Index, list