Skip to content

Commit ae3ed0c

Browse files
authored
Use setup-uv wrapper in upload-benchmark-results (#8078)
## Summary `upload-benchmark-results` calls `astral-sh/setup-uv@v6.1.0` directly without pinning a `version:`, so setup-uv has to query the GitHub API to resolve the latest uv release. In PyTorch CI, setup-uv has been observed rejecting the workflow `GITHUB_TOKEN` (logs `No (valid) GitHub token provided`) and falling back to anonymous requests, which then hit the per-IP rate limit. This is especially painful on the OSDC ARC fleet where many jobs share a NAT egress IP (e.g. `3.139.130.18`), so the 60/hr anonymous budget is exhausted almost immediately. Example failure: https://github.com/pytorch/pytorch/actions/runs/25825809816/job/75882485920 Switch to `pytorch/test-infra/.github/actions/setup-uv`, the wrapper we introduced precisely for this case — it pins uv to an explicit version so no GitHub API call is made. ## Test plan - [ ] After merge, re-run an OSDC ARC test job that uploads benchmark results (e.g. a `linux-jammy-py3.10-clang18` test-osdc shard) and confirm the "Upload the benchmark results" step no longer falls back to anonymous uv setup.
1 parent 9c4ab3f commit ae3ed0c

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

  • .github/actions/upload-benchmark-results

.github/actions/upload-benchmark-results/action.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ runs:
2828
using: composite
2929
steps:
3030
- name: Setup uv
31-
uses: astral-sh/setup-uv@f0ec1fc3b38f5e7cd731bb6ce540c5af426746bb # v6.1.0
31+
uses: pytorch/test-infra/.github/actions/setup-uv@main
3232

3333
- name: Install dependencies
3434
shell: bash

0 commit comments

Comments
 (0)