ReemaAlzaid opened a new pull request, #12345:
URL: https://github.com/apache/gluten/pull/12345
<!--
Thank you for submitting a pull request! Here are some tips:
1. For first-time contributors, please read our contributing guide:
https://github.com/apache/gluten/blob/main/CONTRIBUTING.md
2. If necessary, create a GitHub issue for discussion beforehand to avoid
duplicate work.
3. If the PR is specific to a single backend, include [VL] or [CH] in the PR
title to indicate the
Velox or ClickHouse backend, respectively.
4. If the PR is not ready for review, please mark it as a draft.
-->
## What changes are proposed in this pull request?
Fixes #12280.
Fix Spark 4 Arrow Python UDF execution with the Velox backend by keeping the
Arrow stream writer alive across input batches instead of reopening the IPC
stream per batch.
Also adds a regression test for Arrow Python UDF over Parquet scan
<!--
Provide a clear and concise description of the changes introduced in this PR.
Ensure the PR description aligns with the code changes, especially after
updates.
If applicable, include "Fixes #<GitHub_Issue_ID>" to automatically close the
corresponding issue
when the PR is merged.
-->
## How was this patch tested?
Added `ArrowEvalPythonExecSuite` coverage.
Verified locally on Spark 4.0.2 / Scala 2.13 / linux aarch64. The repro uses
`ColumnarArrowPythonRunner`, returns `max(ship_len) = 7`, and no longer fails
with `Invalid IPC stream`
<!--
Describe how the changes were tested, if applicable.
Include new tests to validate the functionality, if necessary.
For UI-related changes, attach screenshots to demonstrate the updates.
-->
--
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: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
