Skip to content

Use first-party mace interface#497

Open
CompRhys wants to merge 5 commits intomainfrom
first-party-mace
Open

Use first-party mace interface#497
CompRhys wants to merge 5 commits intomainfrom
first-party-mace

Conversation

@CompRhys
Copy link
Member

@CompRhys CompRhys commented Mar 9, 2026

Uses interface introduced in ACEsuit/mace#1390. Awaiting release for minimal pin

@CompRhys CompRhys linked an issue Mar 9, 2026 that may be closed by this pull request
@CompRhys CompRhys requested a review from orionarcher March 10, 2026 18:38
@CompRhys CompRhys marked this pull request as ready for review March 10, 2026 18:38
# dependencies = ["scipy>=1.15", "mace-torch>=0.3.12"]
# dependencies = [
# "torch_sim_atomistic[mace, io]",
# "mace-torch @ git+https://github.com/ACEsuit/mace.git@develop",
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can remove all these once a new mace release comes. I would guess that we should delay to merge this until then but that may not be for a few months

raw_mace_mp = mace_mp(model=MaceUrls.mace_mp_small, return_raw_model=True)
raw_mace_off = mace_off(model=MaceUrls.mace_off_small, return_raw_model=True)
DTYPE = torch.float64
def _download_with_retry(fn: Callable, **kwargs: Any) -> Any:
Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

this didn't help to make the test not flaky.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Move MACE to external model posture

1 participant