Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

GH-43410: [Python] Support Arrow PyCapsule stream objects in write_dataset #43771

Merged

Conversation

jorisvandenbossche
Copy link
Member

@jorisvandenbossche jorisvandenbossche commented Aug 20, 2024

Rationale for this change

Expanding the support internally in pyarrow where we accept objects implementing the Arrow PyCapsule interface. This PR adds support in ds.write_dataset() since we already accept a RecordBatchReader as well.

What changes are included in this PR?

ds.write_dataset() and ds.Scanner.from_baches() now accept any object implementing the Arrow PyCapsule interface for streams.

Are these changes tested?

Yes

Are there any user-facing changes?

No

Copy link

⚠️ GitHub issue #43410 has been automatically assigned in GitHub to PR creator.

Copy link
Member

@kou kou left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

+1

python/pyarrow/tests/test_dataset.py Outdated Show resolved Hide resolved
@github-actions github-actions bot added awaiting merge Awaiting merge and removed awaiting committer review Awaiting committer review labels Aug 20, 2024
@pitrou
Copy link
Member

pitrou commented Sep 18, 2024

@jorisvandenbossche Do you want to update this?

@pitrou pitrou force-pushed the gh-43410-arrow-pycapsule-dataset branch from 7a3f3d5 to d506361 Compare November 18, 2024 15:21
@pitrou
Copy link
Member

pitrou commented Nov 18, 2024

Rebased, will merge if green.

@pitrou pitrou merged commit ad75248 into apache:main Nov 18, 2024
13 checks passed
@pitrou pitrou removed the awaiting merge Awaiting merge label Nov 18, 2024
Copy link

After merging your PR, Conbench analyzed the 3 benchmarking runs that have been run so far on merge-commit ad75248.

There were no benchmark performance regressions. 🎉

The full Conbench report has more details. It also includes information about 5 possible false positives for unstable benchmarks that are known to sometimes produce them.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants