From edd0510c27506ab97f77d7c4010f9acd4b7adc1c Mon Sep 17 00:00:00 2001 From: MorganCThomas Date: Sat, 16 Nov 2024 14:20:12 +0100 Subject: [PATCH] fix acegen to tested versions of torchrl and tensordict; typo --- README.md | 10 +++++++--- requirements.txt | 12 ++++++++++++ tests/test_environment.py | 2 +- 3 files changed, 20 insertions(+), 4 deletions(-) create mode 100644 requirements.txt diff --git a/README.md b/README.md index c75ed75..908b886 100644 --- a/README.md +++ b/README.md @@ -78,14 +78,18 @@ conda create -n acegen python=3.10 -y conda activate acegen ``` -To install the required dependencies run the following commands. Replace `cu121` with your appropriate CUDA version (e.g., `cu118`, `cu117`, `cu102`). +First install `torch` replacing `cu121` with your appropriate CUDA version (e.g., `cu118`, `cu117`, `cu102`). ```bash pip3 install torch torchvision --index-url https://download.pytorch.org/whl/cu121 -pip3 install flake8 pytest pytest-cov hydra-core tqdm wandb -pip3 install torchrl ``` +To install general requirements and the acegen stable versions of `torchrl` and `tensordict` run the following command. + +```bash +pip3 install -r requirements.txt +``` +[Optional] You can install the latest versions by running `pip3 install torchrl --upgrade` but they may not have been tested. diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..75a1960 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,12 @@ +# General dependencies +flake8 +pytest +pytest-cov +hydra-core +tqdm +wandb +rdkit>=2023.3.3 + +# TorchRL and Tensordict +git+https://github.com/pytorch/rl.git@767a877a0a93d41a9b7935598da0ded4c984904f +git+https://github.com/pytorch/tensordict.git@3812ca62d0d5db7d337592d99934e32a3d2b4bfd \ No newline at end of file diff --git a/tests/test_environment.py b/tests/test_environment.py index 5b730e3..6926f16 100644 --- a/tests/test_environment.py +++ b/tests/test_environment.py @@ -18,7 +18,7 @@ skip_if_promptsmiles_not_available = pytest.mark.skipif( not promptsmiles_available, - reason="prompsmiles library is not available, skipping this test", + reason="promptsmiles library is not available, skipping this test", )