Generalise orb version

stfc · Jan 2, 2025 · 061e7ed · 061e7ed
1 parent ba0f18a
commit 061e7ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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