Skip to content

fix(ci): wrap uv install in retry block - #4387

Merged
ko3n1g merged 1 commit into
NVIDIA:mainfrom
ko3n1g:ko3n1g/fix/uv-install-retry
Apr 20, 2026
Merged

fix(ci): wrap uv install in retry block#4387
ko3n1g merged 1 commit into
NVIDIA:mainfrom
ko3n1g:ko3n1g/fix/uv-install-retry

Conversation

@ko3n1g

@ko3n1g ko3n1g commented Apr 20, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Wraps the uv installer curl command in a retry loop (3 attempts, 10 s sleep between attempts), consistent with the existing apt-get retry pattern above it.

Motivation

The install occasionally fails due to transient network issues. Without a retry, the entire CI job fails hard on a flaky download.

Example

for i in 1 2 3; do
  curl -LsSf https://astral.sh/uv/install.sh | UV_INSTALL_DIR=/usr/local/bin sh && break
  echo "uv install attempt $i failed, retrying..."
  sleep 10
done

Signed-off-by: oliver könig <okoenig@nvidia.com>
@ko3n1g
ko3n1g requested a review from a team as a code owner April 20, 2026 09:14
@ko3n1g

ko3n1g commented Apr 20, 2026

Copy link
Copy Markdown
Contributor Author

/ok to test

@svcnvidia-nemo-ci
svcnvidia-nemo-ci marked this pull request as draft April 20, 2026 09:14
@github-actions

Copy link
Copy Markdown
Contributor

This PR has been automatically converted to draft because all PRs must start as drafts.

When you are ready for review, click Ready for Review to begin the review process. This will:

  1. Add the oncall reviewer (optional reviewer)
  2. Add required review teams based on your changes

See the contribution guide for more details.

@copy-pr-bot

copy-pr-bot Bot commented Apr 20, 2026

Copy link
Copy Markdown

Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually.

Contributors can view more details about this message here.

@svcnvidia-nemo-ci svcnvidia-nemo-ci added this to the Core 0.16 milestone Apr 20, 2026
@ko3n1g
ko3n1g marked this pull request as ready for review April 20, 2026 13:35
@svcnvidia-nemo-ci
svcnvidia-nemo-ci requested a review from a team April 20, 2026 13:35
@svcnvidia-nemo-ci svcnvidia-nemo-ci added the Approved All necessary approvals have been made label Apr 20, 2026
@ko3n1g
ko3n1g added this pull request to the merge queue Apr 20, 2026
@svcnvidia-nemo-ci

Copy link
Copy Markdown
Contributor

🔄 Merge queue validation started!

You can track the progress here: https://github.com/NVIDIA/Megatron-LM/actions/runs/24669855535

Merged via the queue into NVIDIA:main with commit 9c210f7 Apr 20, 2026
234 of 237 checks passed
@ko3n1g
ko3n1g deleted the ko3n1g/fix/uv-install-retry branch April 20, 2026 16:09
Phlip79 added a commit to Phlip79/Megatron-LM that referenced this pull request Apr 20, 2026
The rename commit (d6a3854) and subsequent merges accidentally reverted
unrelated upstream changes. Restore upstream versions of:

- megatron/training/arguments.py: restore 'adaptive_muon' optimizer choice
  and --optimizer-cuda-graph argument; remove stray
  --no-scatter-gather-tensors-in-pipeline
- megatron/training/training.py: restore OptimizerCudaGraphWrapper import,
  save_checkpoint_and_time() usage, and timer logic from upstream
- megatron/rl/sequence_packing_utils.py: restore removal of unused
  packed_attention_mask parameter (upstream NVIDIA#3859)
- .github/actions/action.yml: restore retry loop around uv install
  (upstream NVIDIA#4387)

Re-applied only the intended rename-related import path changes to
arguments.py (2 lines) and training.py (2 lines).

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
yangbofun pushed a commit to xlm-research/Megatron-LM that referenced this pull request May 22, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
yhgalaxy pushed a commit to yhgalaxy/Megatron-LM that referenced this pull request Jun 17, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: yhgalaxy <yhgalaxy@outlook.com>
jon-barker pushed a commit to jon-barker/Megatron-LM that referenced this pull request Jul 10, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
Signed-off-by: Jon Barker <jbarker@aws-cmh-slurm-1-vscode-02.cm.cluster>
terminator123 pushed a commit to 021ai/Megatron-LM that referenced this pull request Aug 3, 2026
Signed-off-by: oliver könig <okoenig@nvidia.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Approved All necessary approvals have been made complexity: low Run functional tests

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants