Skip to content

[SPARK-46229][PYTHON][CONNECT] Add applyInArrow to groupBy and cogroup in Spark Connect - #44146

Closed
HyukjinKwon wants to merge 3 commits into
apache:masterfrom
HyukjinKwon:connect-arrow-api
Closed

HyukjinKwon wants to merge 3 commits into
apache:masterfrom
HyukjinKwon:connect-arrow-api

Conversation

@HyukjinKwon

Copy link
Copy Markdown
Member

What changes were proposed in this pull request?

This PR implements Spark Connect version of #38624.

Why are the changes needed?

For feature parity.

Does this PR introduce any user-facing change?

Yes, it adds a new API for Python Spark Connect client.

How was this patch tested?

Reused unittest and doctests.

Was this patch authored or co-authored using generative AI tooling?

No.

@zhengruifeng

Copy link
Copy Markdown
Contributor

LGTM pending CI

@HyukjinKwon HyukjinKwon changed the title [WIP][SPARK-46229][PYTHON][CONNECT] Add applyInArrow to groupBy and cogroup in Spark Connect [SPARK-46229][PYTHON][CONNECT] Add applyInArrow to groupBy and cogroup in Spark Connect Dec 4, 2023
@HyukjinKwon
HyukjinKwon marked this pull request as ready for review December 4, 2023 05:13

@dongjoon-hyun dongjoon-hyun left a comment

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

+1, LGTM.

@HyukjinKwon

Copy link
Copy Markdown
Member Author

I manually tested the latest changes.

Merged to master.


if sys.version_info >= (3, 8):
from typing import Protocol
from typing import Protocol, Tuple

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

nit: This seems to be no-op as Tuple will be overwritten by the following imports?

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

btw, we can remove if sys.version_info >= (3, 8): and the following part as we already dropped Python<3.8?

HyukjinKwon added a commit that referenced this pull request Dec 5, 2023
…ersion check lower than 3.8

### What changes were proposed in this pull request?

This PR addresses the review comment #44146 (comment).

### Why are the changes needed?

We don't support Python lower than 3.8 so we can remove that if-else on Python version.

### Does this PR introduce _any_ user-facing change?

No.

### How was this patch tested?

Existing tests via CI.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes #44168 from HyukjinKwon/SPARK-46229-followup.

Lead-authored-by: Hyukjin Kwon <[email protected]>
Co-authored-by: Hyukjin Kwon <[email protected]>
Signed-off-by: Hyukjin Kwon <[email protected]>
asl3 pushed a commit to asl3/spark that referenced this pull request Dec 5, 2023
…p in Spark Connect

### What changes were proposed in this pull request?

This PR implements Spark Connect version of apache#38624.

### Why are the changes needed?

For feature parity.

### Does this PR introduce _any_ user-facing change?

Yes, it adds a new API for Python Spark Connect client.

### How was this patch tested?

Reused unittest and doctests.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes apache#44146 from HyukjinKwon/connect-arrow-api.

Authored-by: Hyukjin Kwon <[email protected]>
Signed-off-by: Hyukjin Kwon <[email protected]>
asl3 pushed a commit to asl3/spark that referenced this pull request Dec 5, 2023
…ersion check lower than 3.8

### What changes were proposed in this pull request?

This PR addresses the review comment apache#44146 (comment).

### Why are the changes needed?

We don't support Python lower than 3.8 so we can remove that if-else on Python version.

### Does this PR introduce _any_ user-facing change?

No.

### How was this patch tested?

Existing tests via CI.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes apache#44168 from HyukjinKwon/SPARK-46229-followup.

Lead-authored-by: Hyukjin Kwon <[email protected]>
Co-authored-by: Hyukjin Kwon <[email protected]>
Signed-off-by: Hyukjin Kwon <[email protected]>
dbatomic pushed a commit to dbatomic/spark that referenced this pull request Dec 11, 2023
…p in Spark Connect

### What changes were proposed in this pull request?

This PR implements Spark Connect version of apache#38624.

### Why are the changes needed?

For feature parity.

### Does this PR introduce _any_ user-facing change?

Yes, it adds a new API for Python Spark Connect client.

### How was this patch tested?

Reused unittest and doctests.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes apache#44146 from HyukjinKwon/connect-arrow-api.

Authored-by: Hyukjin Kwon <[email protected]>
Signed-off-by: Hyukjin Kwon <[email protected]>
dbatomic pushed a commit to dbatomic/spark that referenced this pull request Dec 11, 2023
…ersion check lower than 3.8

### What changes were proposed in this pull request?

This PR addresses the review comment apache#44146 (comment).

### Why are the changes needed?

We don't support Python lower than 3.8 so we can remove that if-else on Python version.

### Does this PR introduce _any_ user-facing change?

No.

### How was this patch tested?

Existing tests via CI.

### Was this patch authored or co-authored using generative AI tooling?

No.

Closes apache#44168 from HyukjinKwon/SPARK-46229-followup.

Lead-authored-by: Hyukjin Kwon <[email protected]>
Co-authored-by: Hyukjin Kwon <[email protected]>
Signed-off-by: Hyukjin Kwon <[email protected]>
@HyukjinKwon
HyukjinKwon deleted the connect-arrow-api branch January 15, 2024 00:50
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.

4 participants