pre-commit fixes

pandas-dev · Dec 18, 2024 · 7ba4a45 · 7ba4a45
1 parent e46842a
commit 7ba4a45
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 3 deletions.
diff --git a/pandas-stubs/core/strings.pyi b/pandas-stubs/core/strings.pyi
@@ -69,7 +69,12 @@ 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 rsplit(self, pat: str = ..., *, n: int = ..., expand: Literal[True]) -> _TS: ...

diff --git a/tests/test_indexes.py b/tests/test_indexes.py
@@ -114,7 +114,9 @@ def test_str_split() -> None:
     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
+        assert_type(ind.str.split("-", expand=False), "pd.Index[list[str]]"),
+        pd.Index,
+        list,
     )
 
 

diff --git a/tests/test_series.py b/tests/test_series.py
@@ -1557,7 +1557,9 @@ def test_string_accessors():
     # GH 194
     check(assert_type(s.str.split("a", expand=True), pd.DataFrame), pd.DataFrame)
     check(
-        assert_type(s.str.split("a", expand=False), "pd.Series[list[str]]"), pd.Series, list
+        assert_type(s.str.split("a", expand=False), "pd.Series[list[str]]"),
+        pd.Series,
+        list,
     )
     check(assert_type(s.str.startswith("a"), "pd.Series[bool]"), pd.Series, np.bool_)
     check(