Skip to content

Commit

Permalink
Use list of supported backends from top-level __init__.
Browse files Browse the repository at this point in the history
  • Loading branch information
lohedges committed Dec 16, 2024
1 parent 99a1ecc commit e9c9fb8
Showing 1 changed file with 3 additions and 12 deletions.
15 changes: 3 additions & 12 deletions emle/calculator.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,21 +57,12 @@ class EMLECalculator:
Supported backends are listed in the _supported_backends attribute below.
"""

from . import _supported_backends

# Class attributes.

# List of supported backends.
_supported_backends = [
"torchani",
"mace",
"ace",
"deepmd",
"orca",
"rascal",
"sander",
"sqm",
"xtb",
"external",
]
_supported_backends = _supported_backends.copy() + ["external"]

# List of supported devices.
_supported_devices = ["cpu", "cuda"]
Expand Down

0 comments on commit e9c9fb8

Please sign in to comment.