diff --git a/.gitignore b/.gitignore index c1b8c59..fded674 100644 --- a/.gitignore +++ b/.gitignore @@ -161,4 +161,5 @@ cython_debug/ .vscode/ .local/ -.ruff_cache/ \ No newline at end of file +.ruff_cache/ +.idea/ \ No newline at end of file diff --git a/template/steps/promotion/{% if metric_compare_promotion %}promote_get_metrics.py{% endif %} b/template/steps/promotion/{% if metric_compare_promotion %}promote_get_metrics.py{% endif %} index dca8e85..23e7790 100644 --- a/template/steps/promotion/{% if metric_compare_promotion %}promote_get_metrics.py{% endif %} +++ b/template/steps/promotion/{% if metric_compare_promotion %}promote_get_metrics.py{% endif %} @@ -36,7 +36,7 @@ def promote_get_metrics( # Get current model version metric in current run model = get_step_context().model - current_metrics = model.get_model_artifact("model").run_metadata["metrics"].value + current_metrics = model.get_model_artifact("model").run_metadata["metrics"] logger.info(f"Current model version metrics are {current_metrics}") # Get latest saved model version metric in target environment @@ -48,7 +48,7 @@ def promote_get_metrics( except KeyError: latest_version = None if latest_version: - latest_metrics = latest_version.get_model_artifact("model").run_metadata["metrics"].value + latest_metrics = latest_version.get_model_artifact("model").run_metadata["metrics"] logger.info(f"Latest model version metrics are {latest_metrics}") else: logger.info("No currently promoted model version found.")