From fefa3b27654c8e292bf46f282a4b72ed1d8af329 Mon Sep 17 00:00:00 2001 From: Brock Date: Thu, 29 Aug 2024 08:35:31 -0700 Subject: [PATCH] mypy fixup --- pandas/core/arrays/_arrow_string_mixins.py | 10 ++++++---- pandas/core/arrays/string_arrow.py | 1 - 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/pandas/core/arrays/_arrow_string_mixins.py b/pandas/core/arrays/_arrow_string_mixins.py index cd6f2997e8e787..cbe8c66dcbde0f 100644 --- 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 index 8863c1980b60e9..558207af9ce891 100644 --- 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