Skip to content

Commit 9f06042

Browse files
committed
Move fix into _str_contains of ArrowExtensionArray
1 parent 3345bc7 commit 9f06042

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pandas/core/arrays/arrow/array.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2449,19 +2449,23 @@ def _convert_rank_result(self, result):
24492449
def _str_contains(self, pat, case=True, flags=0, na=lib.no_default, regex=True):
24502450
if isinstance(pat, re.Pattern):
24512451
if flags != 0:
2452-
# fallback to python object implementation
24532452
return BaseStringArrayMethods._str_contains(
24542453
self, pat, case, flags, na, regex
24552454
)
24562455
pat = pat.pattern
24572456
regex = True
2457+
elif flags != 0:
2458+
raise NotImplementedError(
2459+
"ArrowExtensionArray does not support str.contains() with flags "
2460+
"for string patterns"
2461+
)
24582462

24592463
try:
24602464
if not regex:
24612465
result = pc.match_substring(self._pa_array, pat, ignore_case=not case)
24622466
else:
24632467
result = pc.match_substring_regex(
2464-
self._pa_array, pat, ignore_case=not case, options=None
2468+
self._pa_array, pat, ignore_case=not case
24652469
)
24662470
return self._convert_bool_result(result, na=na, method_name="contains")
24672471
except (AttributeError, NotImplementedError, pa.ArrowNotImplementedError):

0 commit comments

Comments
 (0)