diff --git a/template/steps/model_promoter.py b/template/steps/model_promoter.py index d00b773..1065c2f 100644 --- 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 )