Skip to content

Commit

Permalink
update to use conda-forge package
Browse files Browse the repository at this point in the history
  • Loading branch information
jthorton committed Feb 1, 2024
1 parent 4758fb4 commit be5e5ba
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 13 deletions.
13 changes: 2 additions & 11 deletions devtools/conda-envs/mace.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -11,18 +11,9 @@ dependencies:
- pydantic>=1.0.0

# mace deps
- pytorch>=2.0.0
- numpy
- scipy
- ase
- opt_einsum
- e3nn
- pymace

# Testing
- pytest
- pytest-cov
- codecov

- pip:
- matscipy
- git+https://github.com/ACEsuit/mace.git@develop
- codecov
4 changes: 2 additions & 2 deletions qcengine/programs/mace.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def found(self, raise_error: bool = False) -> bool:
"mace",
return_bool=True,
raise_error=raise_error,
raise_msg="Please install via github ",
raise_msg="Please install via `mamba install pymace -c conda-forge`",
)

def get_version(self) -> str:
Expand Down Expand Up @@ -65,7 +65,7 @@ def load_model(self, name: str):
model = torch.load(name, map_location=torch.device("cpu"))
except FileNotFoundError:
raise InputError(
"The mace harness can only run local models or a MACE-OFF23 model (`small`, `medium`, `large`"
"The mace harness can only run local models or a MACE-OFF23 model (`small`, `medium`, `large`)"
)
comp_mod = jit.compile(model)
self._CACHE[model_name] = (comp_mod, float(model.r_max), model.atomic_numbers)
Expand Down

0 comments on commit be5e5ba

Please sign in to comment.