put the right versions in deployed models

mir-group · Jun 22, 2023 · 2f43aa8 · 2f43aa8
1 parent 32bad0c
commit 2f43aa8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/nequip/scripts/deploy.py b/nequip/scripts/deploy.py
@@ -24,7 +24,7 @@
 from nequip.model import model_from_config
 from nequip.data import dataset_from_config
 from nequip.utils import Config
-from nequip.utils.versions import check_code_version, get_config_code_versions
+from nequip.utils.versions import check_code_version, get_current_code_versions
 from nequip.scripts.train import default_config
 from nequip.utils.misc import dtype_to_name
 from nequip.utils._global_options import _set_global_options
@@ -287,7 +287,7 @@ def main(args=None):
 
         # Deploy
         metadata: dict = {}
-        code_versions, code_commits = get_config_code_versions(config)
+        code_versions, code_commits = get_current_code_versions(config)
         for code, version in code_versions.items():
             metadata[code + "_version"] = version
         if len(code_commits) > 0: