Skip to content
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.

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

Add upload timeout patch to mlflow on azure #3265

Merged
merged 19 commits into from
May 10, 2024

Conversation

dakinggg
Copy link
Contributor

@dakinggg dakinggg commented May 8, 2024

What does this PR do?

We are fixing an occasional hang during artifact upload by patching mlflow to pass a request timeout along, and to set the default socket timeout in the upload worker function. This will be fixed upstream by mlflow in the next release, at which point we can remove the patch.

Manual tests:

  • upload without the timeout flag set: upload-timeout-test-without-2-xlw9bk
  • upload with the timeout flag set: upload-timeout-test-3-UzoDmx
  • upload with the timeout flag set (and seemingly did something, as this run took longer to complete than the others): upload-timeout-test-2-sVfbkw

@dakinggg dakinggg requested a review from irenedea May 10, 2024 18:47
@dakinggg dakinggg marked this pull request as ready for review May 10, 2024 18:47
@dakinggg dakinggg merged commit d895d56 into mosaicml:dev May 10, 2024
15 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants