From 1ae14e34eebb032075ba0ecdd72e5494a90c1e49 Mon Sep 17 00:00:00 2001
From: Baris Can Durak <bariscandurak@hotmail.com>
Date: Thu, 28 Nov 2024 14:44:40 +0100
Subject: [PATCH] fixed call

---
 template/steps/training/model_trainer.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/template/steps/training/model_trainer.py b/template/steps/training/model_trainer.py
index 2170300..bd5d85b 100644
--- 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 ###