Skip to content

Commit

Permalink
Release 0.3.2 (#188)
Browse files Browse the repository at this point in the history
* Release 0.3.2

* add tabel to explaine how the pt and code versions work together

* emoji for days
  • Loading branch information
mikemhenry authored Sep 22, 2023
1 parent 2e61215 commit 13af90b
Show file tree
Hide file tree
Showing 3 changed files with 21 additions and 10 deletions.
21 changes: 17 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -23,10 +23,10 @@ We show that this approach is not only sufficiently expressive to reproduce lega
# Installation

```bash
$ conda install -c conda-forge -c dglteam "espaloma=0.3.1" "dgl<1"
$ conda install -c conda-forge "espaloma=0.3.2"
```

# Example: Deploy espaloma 0.3.1 pretrained force field to arbitrary MM system
# Example: Deploy espaloma 0.3.2 pretrained force field to arbitrary MM system

```python
# imports
Expand All @@ -51,21 +51,34 @@ espaloma_model(molecule_graph.heterograph)
openmm_system = esp.graphs.deploy.openmm_system_from_graph(molecule_graph)
```

If using espaloma from a local `.pt` file, say for example `espaloma-0.3.1.pt`,
If using espaloma from a local `.pt` file, say for example `espaloma-0.3.2.pt`,
then you would need to run the `eval` method of the model to get the correct
inference/predictions, as follows:

```python
import torch
...
# load local pretrained model
espaloma_model = torch.load("espaloma-0.3.1.pt")
espaloma_model = torch.load("espaloma-0.3.2.pt")
espaloma_model.eval()
...
```

The rest of the code should be the same as in the previous code block example.

# Compatible models

Below is a compatibility matrix for different versions of `espaloma` code and `espaloma` models (the `.pt` file).

| Model 🧪 | DOI 📝 | Supported Espaloma version 💻 | Release Date 🗓️ | Espaloma architecture change 📐? |
|---------------------|-------|------------------------------|----------------|---------------------------------|
| `espaloma-0.3.2.pt` | | 0.3.1, 0.3.2 | Sep 22, 2023 | ✅ No |
| `espaloma-0.3.1.pt` | | 0.3.1, 0.3.2 | Jul 17, 2023 | ⚠️ Yes |
| `espaloma-0.3.0.pt` | | 0.3.0 | Apr 26, 2023 | ⚠️Yes |

> [!NOTE]
> `espaloma-0.3.1.pt` and `espaloma-0.3.2.pt` are the same model.
# Using espaloma to parameterize small molecules in relative free energy calculations

An example of using espaloma to parameterize small molecules in relative alchemical free energy calculations is provided in the `scripts/perses-benchmark/` directory.
Expand Down
6 changes: 3 additions & 3 deletions docs/deploy.rst
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
Deploy espaloma 0.3.1 force field to parametrize your MM system
Deploy espaloma 0.3.2 force field to parametrize your MM system
===============================================================
Pretrained espaloma force field could be deployed on arbitrary small molecule
systems in a few lines::
Expand All @@ -24,12 +24,12 @@ systems in a few lines::
# create an OpenMM System for the specified molecule
openmm_system = esp.graphs.deploy.openmm_system_from_graph(molecule_graph)

If using espaloma from a local ``.pt`` file, say for example ``espaloma-0.3.1.pt``,
If using espaloma from a local ``.pt`` file, say for example ``espaloma-0.3.2.pt``,
then you would need to run the ``eval`` method of the model to get the correct
inference/predictions, as follows::

# load local pretrained model
espaloma_model = torch.load("espaloma-0.3.1.pt")
espaloma_model = torch.load("espaloma-0.3.2.pt")
espaloma_model.eval()

The rest of the code should be the same as in the previous example.
4 changes: 1 addition & 3 deletions docs/install.rst
Original file line number Diff line number Diff line change
Expand Up @@ -4,9 +4,7 @@ Installation
conda
-----

While we are working to put `dgl` on conda-forge, please follow the instructions below to install `espaloma` from `conda-forge` and the package `dgl` package from `dglteam`.

Install espaloma::

conda install -c conda-forge -c dglteam "espaloma=0.3.1" "dgl<1"
conda install -c conda-forge "espaloma=0.3.2"

0 comments on commit 13af90b

Please sign in to comment.