fixed call

zenml-io · Nov 28, 2024 · 1ae14e3 · 1ae14e3
1 parent 8d87577
commit 1ae14e3
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/template/steps/training/model_trainer.py b/template/steps/training/model_trainer.py
@@ -138,6 +138,7 @@ def model_trainer(
     log_metadata(
         metadata={"metrics": eval_results},
         artifact_name="model",
+        infer_artifact=True,
     )
     ### YOUR CODE ENDS HERE ###