Model taken out

htahir1 · htahir1 · commit 58a98b006f6d · 2023-12-21T14:16:41.000+01:00
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
             )

Original file line number	Diff line number	Diff line change
`@@ -46,7 +46,7 @@ def model_promoter(accuracy: float, stage: str = "production") -> bool:`
`46`	`46`	`)`
`47`	`47`	`# We compare their metrics`
`48`	`48`	`prod_accuracy = (`
`49`		`- stage_model_version.get_artifact("model")`
	`49`	`+ stage_model_version.get_artifact("sklearn_classifier")`
`50`	`50`	`.run_metadata["test_accuracy"]`
`51`	`51`	`.value`
`52`	`52`	`)`