Skip to content

tests(functions): add flaky retries for apply axis 1 tests - #18109

Open
shuoweil wants to merge 1 commit into
mainfrom
shuowei-fix-functions-tests
Open

tests(functions): add flaky retries for apply axis 1 tests#18109
shuoweil wants to merge 1 commit into
mainfrom
shuowei-fix-functions-tests

Conversation

@shuoweil

Copy link
Copy Markdown
Contributor

In tests/system/large/functions/test_managed_function.py and tests/system/large/functions/test_remote_function.py, tests exercising DataFrame.apply(axis=1) deploy ephemeral Cloud Functions and BigQuery remote connection assets.

Under parallel CI execution (pytest -n 20), concurrent function deployment requests occasionally hit transient Cloud Functions deployment rate limiting or IAM propagation delays.

Changes

  • Added @pytest.mark.flaky(retries=2, delay=120) to test_df_apply_axis_1_multiple_params, test_df_apply_axis_1_multiple_params_array_output, and test_df_apply_axis_1_single_param_non_series in test_remote_function.py.
  • Added @pytest.mark.flaky(retries=2, delay=30) to test_managed_function_dataframe_apply_axis_1_array_output and test_managed_function_df_apply_axis_1 in test_managed_function.py.

Fixes #<545713683> 🦕

@shuoweil
shuoweil requested a review from sycai August 13, 2026 19:18
@shuoweil shuoweil added the do not merge Indicates a pull request not ready for merge, due to either quality or timing. label Aug 13, 2026
@shuoweil
shuoweil requested review from a team as code owners August 13, 2026 19:18

@gemini-code-assist gemini-code-assist Bot left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

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

Code Review

This pull request adds @pytest.mark.flaky decorators to several system tests in test_managed_function.py and test_remote_function.py to mitigate test flakiness by configuring retries and delays. There are no review comments, so I have no feedback to provide.

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

Labels

do not merge Indicates a pull request not ready for merge, due to either quality or timing.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant