Adjust to zenml-io/zenml#3096

zenml-io · Oct 21, 2024 · a3955d0 · a3955d0
1 parent 0fba996
commit a3955d0
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/.gitignore b/.gitignore
@@ -161,4 +161,5 @@ cython_debug/
 
 .vscode/ 
 .local/
-.ruff_cache/
+.ruff_cache/
+.idea/
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 %}
@@ -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.")