fix naming

zenml-io · Jan 12, 2024 · db7862c · db7862c
1 parent 840a215
commit db7862c
Showing 1 changed file with 3 additions and 3 deletions.
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 = current_version.get_model_artifact("model").run_metadata["metrics"].value
+    current_metrics = model.get_model_artifact("model").run_metadata["metrics"].value
     logger.info(f"Current model version metrics are {current_metrics}")
 
     # Get latest saved model version metric in target environment
@@ -48,8 +48,8 @@ def promote_get_metrics(
     except KeyError:
         latest_version = None
     if latest_version:
-        latest_metrics = current_version.get_model_artifact("model").run_metadata["metrics"].value
-        logger.info(f"Current model version metrics are {latest_metrics}")
+        latest_metrics = latest_version.get_model_artifact("model").run_metadata["metrics"].value
+        logger.info(f"Latest model version metrics are {latest_metrics}")
     else:
         logger.info("No currently promoted model version found.")
         latest_metrics = current_metrics