Merge pull request #55 from Acellera/morgan_dev

update to latest versions of torchrl and tensordict
Acellera · Nov 18, 2024 · 6102f39 · 6102f39
2 parents 5510cc1 + 5854c42
commit 6102f39
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/unittest/install_dependencies.sh b/.github/unittest/install_dependencies.sh
@@ -4,8 +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 git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f
-python -m pip install git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd
+python -m pip install git+https://github.com/pytorch/rl.git@83a7a57daee4b76b249dbc0ce2231f163f2a1482
+python -m pip install git+https://github.com/pytorch/tensordict.git@c11024e0db864bd8487c68e3d0cf5e100fc07831
 
 # Install additional dependencies
 python -m pip install transformers promptsmiles MolScore

diff --git a/requirements.txt b/requirements.txt
@@ -8,5 +8,5 @@ wandb
 rdkit>=2023.3.3
 
 # TorchRL and Tensordict
-git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f
-git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd
+git+https://github.com/pytorch/rl.git@83a7a57daee4b76b249dbc0ce2231f163f2a1482
+git+https://github.com/pytorch/tensordict.git@c11024e0db864bd8487c68e3d0cf5e100fc07831