Skip to content

Commit 76d7a67

Browse files
authored
Update validate_binaries.sh
1 parent 26cb3f6 commit 76d7a67

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

.github/scripts/validate_binaries.sh

+2-2
Original file line numberDiff line numberDiff line change
@@ -60,10 +60,10 @@ conda run -n build_binary pip install torch --index-url "$PYTORCH_URL"
6060
conda run -n build_binary pip install fbgemm-gpu --index-url "$PYTORCH_URL"
6161

6262
# install requirements from pypi
63-
conda run -n build_binary pip install torchmetrics==1.0.3
63+
conda run -n build_binary pip install torchmetrics==1.0.3 tensordict
6464

6565
# install torchrec
66-
conda run -n build_binary pip install torchrec --extra-index-url "$PYTORCH_URL"
66+
conda run -n build_binary pip install torchrec --index-url "$PYTORCH_URL"
6767

6868
# Run small import test
6969
conda run -n build_binary python -c "import torch; import fbgemm_gpu; import torchrec"

0 commit comments

Comments
 (0)