From 95cdda4c77f342f5cd37798d656624028482b462 Mon Sep 17 00:00:00 2001 From: Daniel Heinze Date: Thu, 9 Jan 2020 12:12:44 +0100 Subject: [PATCH] Fixed assignment bug for build_uri If the builduri_base is None, the build_uri would never been instantiated causing an error when calling register_aml_model(model_name, exp, run_id, build_id, build_uri). By adding the else statement, this is prevented. --- code/register/register_model.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/code/register/register_model.py b/code/register/register_model.py index 73b4792e..b2d454bb 100644 --- a/code/register/register_model.py +++ b/code/register/register_model.py @@ -92,7 +92,9 @@ def main(): if (builduri_base is not None): build_uri = builduri_base + build_id run.tag("BuildUri", value=build_uri) - register_aml_model(model_name, exp, run_id, build_id, build_uri) + register_aml_model(model_name, exp, run_id, build_id, build_uri) + else: + register_aml_model(model_name, exp, run_id, build_id) def model_already_registered(model_name, exp, run_id):