Skip to content

Commit

Permalink
Update sevenn (stfc#337)
Browse files Browse the repository at this point in the history
* Update sevenn

* Fix sevenn version
  • Loading branch information
ElliottKasoar authored Oct 23, 2024
1 parent e8ee44c commit 108b5e1
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 3 deletions.
3 changes: 1 addition & 2 deletions janus_core/helpers/mlip_calculators.py
Original file line number Diff line number Diff line change
Expand Up @@ -207,8 +207,7 @@ def choose_calculator(
calculator = AlignnAtomwiseCalculator(path=path, device=device, **kwargs)

elif arch == "sevennet":
# Disable constant-imported-as-non-constant
from sevenn._const import SEVENN_VERSION as __version__ # noqa: N811
from sevenn import __version__
from sevenn.sevennet_calculator import SevenNetCalculator

if isinstance(model_path, Path):
Expand Down
2 changes: 1 addition & 1 deletion pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -45,7 +45,7 @@ alignn = { version = "2024.5.27", optional = true }
chgnet = {version = "0.3.8", optional = true}
dgl = { version = "2.1.0", optional = true } # Pin due to matgl installation issues
matgl = { version = "1.1.3", optional = true}
sevenn = { version = "0.9.3", optional = true }
sevenn = { version = "0.10.0", optional = true }
torchdata = {version = "0.7.1", optional = true} # Pin due to dgl issue
torch_geometric = { version = "^2.5.3", optional = true }
ruff = "^0.5.7"
Expand Down

0 comments on commit 108b5e1

Please sign in to comment.