vyasr opened a new issue, #40780: URL: https://github.com/apache/arrow/issues/40780
### Describe the bug, including details regarding any error messages, version, and platform. `pyarrow.compute.replace_with_mask` requires that the inputs not be chunked arrays. The default calling pattern works as expected: ``` >>> pa.compute.replace_with_mask(pa.array([False]), pa.array([True]), pa.array([True])) <pyarrow.lib.BooleanArray object at 0x7f30ccbc8940> [ true ] ``` while specifying the replacement values as a chunked array raises a comprehensible exception: ``` >>> pa.compute.replace_with_mask(pa.array([False]), pa.array([True]), pa.chunked_array([[True]])) Traceback (most recent call last): File "/home/coder/.conda/envs/rapids/lib/python3.10/code.py", line 90, in runcode exec(code, self.locals) File "<console>", line 1, in <module> File "/home/coder/.conda/envs/rapids/lib/python3.10/site-packages/pyarrow/compute.py", line 246, in wrapper return func.call(args, None, memory_pool) File "pyarrow/_compute.pyx", line 385, in pyarrow._compute.Function.call File "pyarrow/error.pxi", line 154, in pyarrow.lib.pyarrow_internal_check_status File "pyarrow/error.pxi", line 91, in pyarrow.lib.check_status pyarrow.lib.ArrowInvalid: Replacements must be array or scalar, not ChunkedArray([ [ true ] ]) ``` However, specifying the values as a chunked array instead silently produces an invalid output: ``` >>> pa.compute.replace_with_mask(pa.chunked_array([[False]]), pa.array([True]), pa.array([True])) <pyarrow.lib.ChunkedArray object at 0x7f30ccbcc5e0> [ <Invalid array: Buffer #1 too small in array of type bool and length 1: expected at least 1 byte(s), got 0> ] ``` The latter case should also be handled by the same error-checking logic used to validate the replacements (unless chunked arrays can in fact be supported, which would be nice but isn't too important). ### Component(s) Python -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@arrow.apache.org.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org