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")