fixup

Author: MarcoGorelli

pandas-stubs/core/strings.pyi

@@ -95,19 +95,19 @@ class StringMethods(NoNewAttributesMixin, Generic[T, _TS, _TM, _TS2, _TI, _TE, _
         self, pat: str = ..., *, n: int = ..., expand: Literal[False] = ...
     ) -> _TS2: ...
     @overload
-    def partition(self, sep: str = ...) -> pd.DataFrame: ...
+    def partition(self, sep: str = ...) -> _TS: ...
     @overload
-    def partition(self, *, expand: Literal[True]) -> pd.DataFrame: ...
+    def partition(self, *, expand: Literal[True]) -> _TS: ...
     @overload
-    def partition(self, sep: str, expand: Literal[True]) -> pd.DataFrame: ...
+    def partition(self, sep: str, expand: Literal[True]) -> _TS: ...
     @overload
     def partition(
         self, sep: str, expand: Literal[False]
     ) -> pd.Series[type[object]]: ...
     @overload
     def partition(self, *, expand: Literal[False]) -> pd.Series[type[object]]: ...
     @overload
-    def rpartition(self, sep: str = ...) -> pd.DataFrame: ...
+    def rpartition(self, sep: str = ...) -> _TS: ...
     @overload
     def rpartition(self, *, expand: Literal[True]) -> pd.DataFrame: ...
     @overload
@@ -171,7 +171,7 @@ class StringMethods(NoNewAttributesMixin, Generic[T, _TS, _TM, _TS2, _TI, _TE, _
         break_long_words: bool | None = ...,
         break_on_hyphens: bool | None = ...,
     ) -> T: ...
-    def get_dummies(self, sep: str = ...) -> pd.DataFrame: ...
+    def get_dummies(self, sep: str = ...) -> _TS: ...
     def translate(self, table: dict[int, int | str | None] | None) -> T: ...
     def count(self, pat: str, flags: int = ...) -> _TI: ...
     def startswith(self, pat: str | tuple[str, ...], na: Any = ...) -> _TM: ...
@@ -180,7 +180,7 @@ class StringMethods(NoNewAttributesMixin, Generic[T, _TS, _TM, _TS2, _TI, _TE, _
     @overload
     def extract(
         self, pat: str, flags: int = ..., *, expand: Literal[True] = ...
-    ) -> pd.DataFrame: ...
+    ) -> _TS: ...
     @overload
     def extract(
         self, pat: str, flags: int, expand: Literal[False]
@@ -189,7 +189,7 @@ class StringMethods(NoNewAttributesMixin, Generic[T, _TS, _TM, _TS2, _TI, _TE, _
     def extract(
         self, pat: str, flags: int = ..., *, expand: Literal[False]
     ) -> Series[type[object]]: ...
-    def extractall(self, pat: str, flags: int = ...) -> pd.DataFrame: ...
+    def extractall(self, pat: str, flags: int = ...) -> _TS: ...
     def find(self, sub: str, start: int = ..., end: int | None = ...) -> _TI: ...
     def rfind(self, sub: str, start: int = ..., end: int | None = ...) -> _TI: ...
     def normalize(self, form: Literal["NFC", "NFKC", "NFD", "NFKD"]) -> T: ...