Skip to content

feat: retry AI/ML jobs that fail more often #1965

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
merged 10 commits into from
Aug 5, 2025
Merged

feat: retry AI/ML jobs that fail more often #1965

merged 10 commits into from
Aug 5, 2025

Conversation

tswast
Copy link
Collaborator

@tswast tswast commented Aug 5, 2025

Thank you for opening a Pull Request! Before submitting your PR, there are a few things you can do to make sure it goes smoothly:

  • Make sure to open an issue as a bug/issue before writing your code! That way we can discuss the change, evaluate designs, and agree on the general idea
  • Ensure the tests and linter pass
  • Code coverage does not decrease (if any source code was changed)
  • Appropriate docs were updated (if necessary)

Fixes internal issue b/436586523 🦕

@tswast tswast requested review from a team as code owners August 5, 2025 20:21
@tswast tswast requested a review from TrevorBergeron August 5, 2025 20:21
@product-auto-label product-auto-label bot added the size: m Pull request size is medium. label Aug 5, 2025
@product-auto-label product-auto-label bot added the api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. label Aug 5, 2025
tswast
tswast commented Aug 5, 2025
View reviewed changes
tswast
tswast commented Aug 5, 2025
View reviewed changes
tswast and others added 8 commits August 5, 2025 15:23
@tswast
Update pyproject.toml
f7e3e36
@gcf-owl-bot
🦉 Updates from OwlBot post-processor
b8c5da8 
See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
@gcf-owl-bot
Merge branch 'b436586523-ml-ai-job_retry' of https://github.com/googl…
1a83355 
…eapis/python-bigquery-dataframes into b436586523-ml-ai-job_retry
@gcf-owl-bot
🦉 Updates from OwlBot post-processor
4d223cd 
See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
@gcf-owl-bot
Merge branch 'b436586523-ml-ai-job_retry' of https://github.com/googl…
e46a718 
…eapis/python-bigquery-dataframes into b436586523-ml-ai-job_retry
@gcf-owl-bot
🦉 Updates from OwlBot post-processor
bd5f737 
See https://github.com/googleapis/repo-automation-bots/blob/main/packages/owl-bot/README.md
@gcf-owl-bot
Merge branch 'b436586523-ml-ai-job_retry' of https://github.com/googl…
8bdbcb2 
…eapis/python-bigquery-dataframes into b436586523-ml-ai-job_retry
@tswast
fix mypy
800aff9
@product-auto-label product-auto-label bot added size: l Pull request size is large. and removed size: m Pull request size is medium. labels Aug 5, 2025
@tswast
Copy link
Collaborator Author

tswast commented Aug 5, 2025

notebook tests are running, but I can see that bq_dataframes_llm_kmeans.ipynb passed already in the logs. That's a good sign!

@tswast
Copy link
Collaborator Author

tswast commented Aug 5, 2025

e2e failure is on unit_prerelease, not related to this change:

name = 'polars', import_ = <function _gcd_import at 0x7f3c21a284a0>

>   ???
E   ModuleNotFoundError: No module named 'polars'

@tswast tswast merged commit 25bde9f into main Aug 5, 2025
19 of 25 checks passed
@tswast tswast deleted the b436586523-ml-ai-job_retry branch August 5, 2025 21:19
@release-please release-please bot mentioned this pull request Aug 5, 2025
Merged
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@shuoweil shuoweil shuoweil approved these changes

@TrevorBergeron TrevorBergeron Awaiting requested review from TrevorBergeron TrevorBergeron is a code owner automatically assigned from googleapis/api-bigquery-dataframe

Assignees

@jialuoo jialuoo

Labels
api: bigquery Issues related to the googleapis/python-bigquery-dataframes API. size: l Pull request size is large.
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@tswast @shuoweil @jialuoo