diff --git a/janus_core/helpers/mlip_calculators.py b/janus_core/helpers/mlip_calculators.py index 6780ee4e..a1310c2c 100644 --- a/janus_core/helpers/mlip_calculators.py +++ b/janus_core/helpers/mlip_calculators.py @@ -226,7 +226,7 @@ def choose_calculator( calculator = SevenNetCalculator(model=model_path, device=device, **kwargs) elif arch == "orb": - __version__ = "0.3" + from orb_models import __version__ from orb_models.forcefield.calculator import ORBCalculator from orb_models.forcefield.graph_regressor import GraphRegressor import orb_models.forcefield.pretrained as orb_ff