Skip to content

Commit cb27dd6

Browse files
starlord-danieleedorenko
authored andcommitted
Fixed assignment bug for build_uri (#133)
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.
1 parent dc88c72 commit cb27dd6

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

code/register/register_model.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -92,7 +92,9 @@ def main():
9292
if (builduri_base is not None):
9393
build_uri = builduri_base + build_id
9494
run.tag("BuildUri", value=build_uri)
95-
register_aml_model(model_name, exp, run_id, build_id, build_uri)
95+
register_aml_model(model_name, exp, run_id, build_id, build_uri)
96+
else:
97+
register_aml_model(model_name, exp, run_id, build_id)
9698

9799

98100
def model_already_registered(model_name, exp, run_id):

0 commit comments

Comments
 (0)