Skip to content

Conversation

@zecookiez
Copy link
Contributor

What changes were proposed in this pull request?

SPARK-51252

Fix for flaky tests described in #50030.

This PR relaxes the version check in the tests and processes more data in queries to force more batches.

I also reduced the number of shuffle partitions for one of the join queries because there used to be 5 * 4 state stores that would need a snapshot to upload in a short time span. Now there's only 3 * 4 stores to deal with.

Why are the changes needed?

The tests previously introduced in #50030 are quite flaky depending on the environment, resulting in CI gates failing some tests in the suite.

Does this PR introduce any user-facing change?

No.

How was this patch tested?

Re-ran and verified the tests multiple times locally, but primarily using the current merge gates to see if these pass properly.

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

No.

@HeartSaVioR HeartSaVioR changed the title [SPARK-51252][SS] Make tests more consistent for StateStoreInstanceMetricSuite [SPARK-51252][SS][TESTS] Make tests more consistent for StateStoreInstanceMetricSuite Mar 26, 2025
Copy link
Contributor

@HeartSaVioR HeartSaVioR left a comment

Choose a reason for hiding this comment

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

+1
Thanks for the quick fix!

@HeartSaVioR
Copy link
Contributor

HeartSaVioR commented Mar 26, 2025

I'll merge this tomorrow to give others a chance to review it and leave any major comment.

@HeartSaVioR
Copy link
Contributor

Thanks! Merging to master.

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