From 445203f4ec60900cdf9093d5fb4f861424d08b8c Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Fri, 18 Mar 2022 10:01:21 -0400 Subject: [PATCH] ArrowStringArray._cmp_method TODO --- pandas/core/arrays/string_arrow.py | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/pandas/core/arrays/string_arrow.py b/pandas/core/arrays/string_arrow.py index e39ebd3afd2ff..eb3b3b8d29647 100644 --- a/pandas/core/arrays/string_arrow.py +++ b/pandas/core/arrays/string_arrow.py @@ -332,8 +332,11 @@ def _cmp_method(self, other, op): else: return NotImplemented - # TODO(ARROW-9429): Add a .to_numpy() to ChunkedArray - return BooleanArray._from_sequence(result.to_pandas().values) + if pa_version_under2p0: + result = result.to_pandas().values + else: + result = result.to_numpy() + return BooleanArray._from_sequence(result) def insert(self, loc: int, item): if not isinstance(item, str) and item is not libmissing.NA: