Skip to content

[ML] Run PyTorch allowlist validation as a dedicated Buildkite step#3010

Open
edsavage wants to merge 1 commit intoelastic:mainfrom
edsavage:fix/pytorch-validation-dedicated-step
Open

[ML] Run PyTorch allowlist validation as a dedicated Buildkite step#3010
edsavage wants to merge 1 commit intoelastic:mainfrom
edsavage:fix/pytorch-validation-dedicated-step

Conversation

@edsavage
Copy link
Contributor

@edsavage edsavage commented Mar 24, 2026

Summary

  • The PyTorch allowlist validation was inlined in run_tests.sh, but the Linux Docker build/test images don't include Python 3 (it's used during image builds to compile PyTorch, then dropped in the multi-stage final image). This caused the validation to always skip with OPTIONAL=TRUE or fail without it.
  • Moves the validation to a dedicated pipeline step using a python:3 agent image, triggered only for run_pytorch_tests builds. Runs in parallel with other post-build steps and soft-fails.

Test plan

  • Trigger buildkite run_pytorch_tests and verify the "Validate PyTorch allowlist" step appears and runs in a python:3 container
  • Verify normal PR builds don't include the validation step

Made with Cursor

The validation was previously inlined in run_tests.sh, but the Linux
build/test Docker images don't include Python 3 (it's only used
during image builds to compile PyTorch, then dropped in the
multi-stage final image).

Move the validation to a dedicated pipeline step using a python:3
agent image, triggered only for run_pytorch_tests builds. This runs
in parallel with the other post-build steps and soft-fails so it
doesn't block the build.

Made-with: Cursor
@prodsecmachine
Copy link

prodsecmachine commented Mar 24, 2026

Snyk checks have passed. No issues have been found so far.

Status Scan Engine Critical High Medium Low Total (0)
Open Source Security 0 0 0 0 0 issues
Licenses 0 0 0 0 0 issues

💻 Catch issues earlier using the plugins for VS Code, JetBrains IDEs, Visual Studio, and Eclipse.

@edsavage
Copy link
Contributor Author

buildkite run_pytorch_tests

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