From 97c36680f2c31f76e3348362c51aa10e8ea50358 Mon Sep 17 00:00:00 2001 From: MorganCThomas Date: Mon, 18 Nov 2024 10:38:43 +0100 Subject: [PATCH] make everything explicit, was downloading the wrong version fo torchrl --- .github/unittest/install_dependencies.sh | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh index 6660260..56e1f17 100644 --- a/.github/unittest/install_dependencies.sh +++ b/.github/unittest/install_dependencies.sh @@ -4,7 +4,8 @@ python -m pip install --upgrade pip # Install required dependencies python -m pip install flake8 pytest pytest-cov hydra-core tqdm python -m pip install torch torchvision -python -m pip install -r acegen-open/requirements.txt +python -m pip install git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f +python -m pip install git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd # Install additional dependencies python -m pip install transformers promptsmiles MolScore