mypy fixup

pandas-dev · Aug 29, 2024 · fefa3b2 · fefa3b2
1 parent a6abf99
commit fefa3b2
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/pandas/core/arrays/_arrow_string_mixins.py b/pandas/core/arrays/_arrow_string_mixins.py
@@ -156,16 +156,18 @@ def _str_find(self, sub: str, start: int = 0, end: int | None = None):
             and not (start == 0 and end is None)
         ):
             # GH#59562
-            result = self._apply_elementwise(lambda val: val.find(sub, start, end))
-            return self._convert_int_result(pa.chunked_array(result))
+            res_list = self._apply_elementwise(lambda val: val.find(sub, start, end))
+            return self._convert_int_result(pa.chunked_array(res_list))
 
         if (start == 0 or start is None) and end is None:
             result = pc.find_substring(self._pa_array, sub)
         else:
             if sub == "":
                 # GH#56792
-                result = self._apply_elementwise(lambda val: val.find(sub, start, end))
-                return self._convert_int_result(pa.chunked_array(result))
+                res_list = self._apply_elementwise(
+                    lambda val: val.find(sub, start, end)
+                )
+                return self._convert_int_result(pa.chunked_array(res_list))
             if start is None:
                 start_offset = 0
                 start = 0

diff --git a/pandas/core/arrays/string_arrow.py b/pandas/core/arrays/string_arrow.py
@@ -282,7 +282,6 @@ def astype(self, dtype, copy: bool = True):
     _str_map = BaseStringArray._str_map
     _str_startswith = ArrowStringArrayMixin._str_startswith
     _str_endswith = ArrowStringArrayMixin._str_endswith
-    _str_find = ArrowExtensionArray._str_find
 
     def _str_contains(
         self, pat, case: bool = True, flags: int = 0, na=np.nan, regex: bool = True