Model taken out

zenml-io · Dec 21, 2023 · 58a98b0 · 58a98b0
1 parent 56360c9
commit 58a98b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/template/steps/model_promoter.py b/template/steps/model_promoter.py
@@ -46,7 +46,7 @@ def model_promoter(accuracy: float, stage: str = "production") -> bool:
             )
             # We compare their metrics
             prod_accuracy = (
-                stage_model_version.get_artifact("model")
+                stage_model_version.get_artifact("sklearn_classifier")
                 .run_metadata["test_accuracy"]
                 .value
             )