Skip to content

Commit

Permalink
fix acegen to tested versions of torchrl and tensordict; typo
Browse files Browse the repository at this point in the history
  • Loading branch information
MorganCThomas committed Nov 16, 2024
1 parent 551ba16 commit edd0510
Show file tree
Hide file tree
Showing 3 changed files with 20 additions and 4 deletions.
10 changes: 7 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -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.

</details>

Expand Down
12 changes: 12 additions & 0 deletions requirements.txt
Original file line number Diff line number Diff line change
@@ -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
2 changes: 1 addition & 1 deletion tests/test_environment.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
)


Expand Down

0 comments on commit edd0510

Please sign in to comment.