Skip to content

Conversation

@ueshin
Copy link
Member

@ueshin ueshin commented Nov 25, 2025

What changes were proposed in this pull request?

Re-enables tests in test_parity_udf_profiler and test_parity_memory_profiler.

Added utility decorators for Spark Connect parity tests.

  • skip_if_server_version_is
  • skip_if_server_version_is_greater_than_or_equal_to

Why are the changes needed?

Some tests are disabled due to Spark Connect cross version tests.

Does this PR introduce any user-facing change?

No.

How was this patch tested?

Modified the related tests to use the utility.

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

No.

@ueshin ueshin changed the title [SPARK-54485][SPARK-54494][TESTS] Enable tests in test_parity_udf_profiler and test_parity_memory_profiler [SPARK-54485][SPARK-54494][TESTS] Re-enable tests in test_parity_udf_profiler and test_parity_memory_profiler Nov 25, 2025
@github-actions github-actions bot added the CORE label Nov 25, 2025
return contextlib.nullcontext()


def skip_if_server_version_is(
Copy link
Contributor

Choose a reason for hiding this comment

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

this seems to be a helper decorator. should we make it a private one?

Copy link
Member Author

@ueshin ueshin Nov 25, 2025

Choose a reason for hiding this comment

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

What do you mean by "make it a private one"?
I don't intend it to be a private as it's complete usable, but I'm ok to make it a private. WDYT? @HyukjinKwon

Copy link
Member

Choose a reason for hiding this comment

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

It's fine. let's just leave it

@ueshin
Copy link
Member Author

ueshin commented Nov 25, 2025

If the decorators look good, I'll submit a PR against master to forward-port them.

Copy link
Member

@HyukjinKwon HyukjinKwon left a comment

Choose a reason for hiding this comment

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

Oh, I like this. We should put the util in master too

@ueshin ueshin changed the title [SPARK-54485][SPARK-54494][TESTS] Re-enable tests in test_parity_udf_profiler and test_parity_memory_profiler [SPARK-54485][SPARK-54494][PYTHON][TESTS] Re-enable tests in test_parity_udf_profiler and test_parity_memory_profiler Nov 25, 2025
@ueshin
Copy link
Member Author

ueshin commented Nov 25, 2025

#53223 for forward-port.

dongjoon-hyun pushed a commit that referenced this pull request Nov 26, 2025
…ct parity tests

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

This PR aims to add utility decorators in Spark Connect cross version tests.

- `skip_if_server_version_is`
- `skip_if_server_version_is_greater_than_or_equal_to`

### Why are the changes needed?

This is a forward-port from #53222.

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

No.

### How was this patch tested?

N/A

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

No.

Closes #53223 from ueshin/issues/SPARK-54517/decorators.

Authored-by: Takuya Ueshin <ueshin@databricks.com>
Signed-off-by: Dongjoon Hyun <dongjoon@apache.org>
dongjoon-hyun pushed a commit that referenced this pull request Nov 26, 2025
…ct parity tests

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

This PR aims to add utility decorators in Spark Connect cross version tests.

- `skip_if_server_version_is`
- `skip_if_server_version_is_greater_than_or_equal_to`

### Why are the changes needed?

This is a forward-port from #53222.

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

No.

### How was this patch tested?

N/A

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

No.

Closes #53223 from ueshin/issues/SPARK-54517/decorators.

Authored-by: Takuya Ueshin <ueshin@databricks.com>
Signed-off-by: Dongjoon Hyun <dongjoon@apache.org>
(cherry picked from commit d9dcb5c)
Signed-off-by: Dongjoon Hyun <dongjoon@apache.org>
@dongjoon-hyun dongjoon-hyun changed the title [SPARK-54485][SPARK-54494][PYTHON][TESTS] Re-enable tests in test_parity_udf_profiler and test_parity_memory_profiler [SPARK-54485][SPARK-54494][PYTHON][TESTS][4.0] Re-enable tests in test_parity_udf_profiler and test_parity_memory_profiler Nov 26, 2025
Copy link
Member

@dongjoon-hyun dongjoon-hyun left a comment

Choose a reason for hiding this comment

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

I merged the utility function PRs to master/4.1.

@ueshin
Copy link
Member Author

ueshin commented Nov 26, 2025

The failed tests are Run / Build modules: pyspark-mllib, pyspark-ml, pyspark-ml-connect.
They are not related to this PR, and can't pass on branch-4.0 in my GitHub Action for some reason same as before.

Retrying just in case:

@HyukjinKwon
Copy link
Member

That's fine.

Merged to branch-4.0.

HyukjinKwon pushed a commit that referenced this pull request Nov 26, 2025
…t_parity_udf_profiler and test_parity_memory_profiler

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

Re-enables tests in `test_parity_udf_profiler` and `test_parity_memory_profiler`.

Added utility decorators for Spark Connect parity tests.

- `skip_if_server_version_is`
- `skip_if_server_version_is_greater_than_or_equal_to`

### Why are the changes needed?

Some tests are disabled due to Spark Connect cross version tests.

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

No.

### How was this patch tested?

Modified the related tests to use the utility.

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

No.

Closes #53222 from ueshin/issues/SPARK-54485_54494/profiler_tests.

Authored-by: Takuya Ueshin <ueshin@databricks.com>
Signed-off-by: Hyukjin Kwon <gurwls223@apache.org>
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.

5 participants