anishshri-db opened a new pull request, #50626: URL: https://github.com/apache/spark/pull/50626
### What changes were proposed in this pull request? Fix test case related to extra batch causing empty df due to watermark ### Why are the changes needed? Without this change, the test can be flaky due to the extra batch ### Does this PR introduce _any_ user-facing change? No ### How was this patch tested? ``` Will test the following Python tests: ['pyspark.sql.tests.pandas.test_pandas_transform_with_state TransformWithStateInPandasTests.test_transform_with_state_with_wmark_and_non_event_time'] python3.9 python_implementation is CPython python3.9 version is: Python 3.9.20 Starting test(python3.9): pyspark.sql.tests.pandas.test_pandas_transform_with_state TransformWithStateInPandasTests.test_transform_with_state_with_wmark_and_non_event_time (temp output: /Users/anish.shrigondekar/spark/spark/python/target/256cf6e3-872e-4590-b7d9-26677e48c48f/python3.9__pyspark.sql.tests.pandas.test_pandas_transform_with_state_TransformWithStateInPandasTests.test_transform_with_state_with_wmark_and_non_event_time__co1fbfji.log) Finished test(python3.9): pyspark.sql.tests.pandas.test_pandas_transform_with_state TransformWithStateInPandasTests.test_transform_with_state_with_wmark_and_non_event_time (51s) Tests passed in 51 seconds ``` ### Was this patch authored or co-authored using generative AI tooling? No -- 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: reviews-unsubscr...@spark.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org