From 455d976d02c6c08dd88a273d2b85c82bb7795ac8 Mon Sep 17 00:00:00 2001
From: Huanyu He <hhy@meta.com>
Date: Thu, 20 Feb 2025 16:57:40 -0800
Subject: [PATCH] adding tensordict dependency in validate_binaries.sh (#2756)

Summary:
Pull Request resolved: https://github.com/pytorch/torchrec/pull/2756

# context
* failed to directly install torchrec from nightly build: P1736807637
* the reason is that tensordict is not available from https://download.pytorch.org/whl/nightly/*
* so the tensordict has to be installed separately

# changes
* install current tensordict version in the validate_binaries.sh script

Reviewed By: dstaay-fb

Differential Revision: D69950708

fbshipit-source-id: 2afc83eccc8169f0bd80d3d117059f914d3e16b2
---
 .github/scripts/validate_binaries.sh | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/.github/scripts/validate_binaries.sh b/.github/scripts/validate_binaries.sh
index f9958a208..076672bbb 100755
--- a/.github/scripts/validate_binaries.sh
+++ b/.github/scripts/validate_binaries.sh
@@ -80,6 +80,9 @@ conda run -n "${CONDA_ENV}" pip install fbgemm-gpu --index-url "$PYTORCH_URL"
 # install requirements from pypi
 conda run -n "${CONDA_ENV}" pip install torchmetrics==1.0.3
 
+# install tensordict from pypi
+conda run -n "${CONDA_ENV}" pip install tensordict==0.7.1
+
 # install torchrec
 conda run -n "${CONDA_ENV}" pip install torchrec --index-url "$PYTORCH_URL"