[ARROW-15928] [C++][Python] replace_with_mask seg faults when passed ChunkedArray

• Type: Bug
• Status: Resolved
• Priority: Critical
• Resolution: Fixed
• Affects Version/s: 5.0.0, 6.0.0, 6.0.1, 7.0.0
• Fix Version/s: 8.0.0
• Component/s: C++, Python
• Labels: good-second-issue kernel pull-request-available
• External issue URL: https://github.com/apache/arrow/issues/31356

Language

Python

• Assignee: David Li
• Reporter: Luke Manley
• Created: 14/Mar/22 13:01
• Updated: 11/Jan/23 11:40
• Resolved: 25/Apr/22 14:38

Description

It looks like most compute functions that take an array-like can accept a ChunkedArray as well. One exception to that appears to be replace_with_mask which seems to seg fault when the array-like is a chunked array. Here is an example:
 
 

import pyarrow as pa 
import pyarrow.compute as pc 

ca = pa.chunked_array([[1, 2], [3, 4]]) 
mask = [False, True, False, True] 

# works (when we first combine chunks into a single array) 
a_new = pc.replace_with_mask(ca.combine_chunks(), mask, 0) 

# seg fault (if we try to pass the chunked array) 
ca_new = pc.replace_with_mask(ca, mask, 0)

 

Issue Links

• is related to: ARROW-16271 [C++] Implement full chunked array support for replace_with_mask
• links to: GitHub Pull Request #12802