From 2f43aa84542df733bbe38cb9d6cca176b0e98054 Mon Sep 17 00:00:00 2001 From: Linux-cpp-lisp <1473644+Linux-cpp-lisp@users.noreply.github.com> Date: Thu, 22 Jun 2023 19:27:25 -0400 Subject: [PATCH] put the right versions in deployed models --- nequip/scripts/deploy.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/nequip/scripts/deploy.py b/nequip/scripts/deploy.py index e90466c2..bcee8255 100644 --- 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: