From b943e9ca3c4b75b02fa8afbf22a84b438fea60ca Mon Sep 17 00:00:00 2001 From: Roman Fenioux Date: Thu, 30 Oct 2025 11:19:19 +0100 Subject: [PATCH] Add description metadata to model info --- ci/update_model_info_deparate.py | 3 ++- ci/update_model_info_deparate_ngc.py | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/ci/update_model_info_deparate.py b/ci/update_model_info_deparate.py index 5f4756f9..a28e73c8 100644 --- a/ci/update_model_info_deparate.py +++ b/ci/update_model_info_deparate.py @@ -84,10 +84,11 @@ def update_model_info( model_info = get_json_dict(model_info_path) if bundle_name_with_version not in model_info.keys(): - model_info[bundle_name_with_version] = {"checksum": "", "source": ""} + model_info[bundle_name_with_version] = {"checksum": "", "source": "", "description": ""} model_info[bundle_name_with_version]["checksum"] = checksum model_info[bundle_name_with_version]["source"] = source + model_info[bundle_name_with_version]["description"] = metadata["description"] save_model_info(model_info, model_info_path) return (True, "update successful") diff --git a/ci/update_model_info_deparate_ngc.py b/ci/update_model_info_deparate_ngc.py index decf6c66..d0ea4e07 100644 --- a/ci/update_model_info_deparate_ngc.py +++ b/ci/update_model_info_deparate_ngc.py @@ -94,10 +94,11 @@ def update_model_info( # step 4 if bundle_name_with_version not in model_info.keys(): - model_info[bundle_name_with_version] = {"checksum": "", "source": ""} + model_info[bundle_name_with_version] = {"checksum": "", "source": "", "description": ""} model_info[bundle_name_with_version]["checksum"] = checksum model_info[bundle_name_with_version]["source"] = source + model_info[bundle_name_with_version]["description"] = metadata["description"] save_model_info(model_info, model_info_path) return (True, "update successful")