diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh index 85ad0de47..a7dac3146 100755 --- a/.github/scripts/validate_binaries.sh +++ b/.github/scripts/validate_binaries.sh @@ -60,7 +60,7 @@ conda run -n build_binary pip install torch --index-url "$PYTORCH_URL" conda run -n build_binary pip install fbgemm-gpu --index-url "$PYTORCH_URL" # install requirements from pypi -conda run -n build_binary pip install torchmetrics==1.0.3 +conda run -n build_binary pip install torchmetrics==1.0.3 tensordict # install torchrec conda run -n build_binary pip install torchrec --index-url "$PYTORCH_URL" diff --git a/.github/workflows/validate-nightly-binaries.yml b/.github/workflows/validate-nightly-binaries.yml index 6d6369495..eaab8bc93 100644 --- a/.github/workflows/validate-nightly-binaries.yml +++ b/.github/workflows/validate-nightly-binaries.yml @@ -19,6 +19,7 @@ on: - .github/workflows/validate-nightly-binaries.yml - .github/workflows/validate-binaries.yml - .github/scripts/validate-binaries.sh + jobs: nightly: uses: ./.github/workflows/validate-binaries.yml