Skip to content

Commit 061e7ed

Browse files
committed
Generalise orb version
1 parent ba0f18a commit 061e7ed

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

janus_core/helpers/mlip_calculators.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -226,7 +226,7 @@ def choose_calculator(
226226
calculator = SevenNetCalculator(model=model_path, device=device, **kwargs)
227227

228228
elif arch == "orb":
229-
__version__ = "0.3"
229+
from orb_models import __version__
230230
from orb_models.forcefield.calculator import ORBCalculator
231231
from orb_models.forcefield.graph_regressor import GraphRegressor
232232
import orb_models.forcefield.pretrained as orb_ff

0 commit comments

Comments
 (0)