From f2d82002524719a1bd5637a3f2a44132c00a743a Mon Sep 17 00:00:00 2001 From: David Tesar Date: Mon, 25 Nov 2019 18:27:50 -0800 Subject: [PATCH 1/2] Fix validate logic for no model found --- code/register/register_model.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/code/register/register_model.py b/code/register/register_model.py index d3e7e607..634d3f7f 100644 --- a/code/register/register_model.py +++ b/code/register/register_model.py @@ -109,8 +109,12 @@ def main(): if (validate): try: tag_name = 'BuildId' - get_model_by_tag(model_name, tag_name, build_id, exp.workspace) - print("Model was registered for this build.") + model = get_model_by_tag(model_name, tag_name, build_id, exp.workspace) + if (model is not None): + print("Model was registered for this build.") + if (model is None): + print("Model was not registered for this run.") + sys.exit(1) except Exception as e: print(e) print("Model was not registered for this run.") From f5e75b21d7c983a6333ff3888c09e8d68eff20cb Mon Sep 17 00:00:00 2001 From: David Tesar Date: Mon, 25 Nov 2019 18:30:13 -0800 Subject: [PATCH 2/2] Fix lint --- code/register/register_model.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/code/register/register_model.py b/code/register/register_model.py index 634d3f7f..a1b95696 100644 --- a/code/register/register_model.py +++ b/code/register/register_model.py @@ -109,7 +109,8 @@ def main(): if (validate): try: tag_name = 'BuildId' - model = get_model_by_tag(model_name, tag_name, build_id, exp.workspace) + model = get_model_by_tag( + model_name, tag_name, build_id, exp.workspace) if (model is not None): print("Model was registered for this build.") if (model is None):