fixed call

bcdurak · bcdurak · commit 1ae14e34eebb · 2024-11-28T14:44:40.000+01:00
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 ###
 

Original file line number	Diff line number	Diff line change
`@@ -138,6 +138,7 @@ def model_trainer(`
`138`	`138`	`log_metadata(`
`139`	`139`	`metadata={"metrics": eval_results},`
`140`	`140`	`artifact_name="model",`
	`141`	`+ infer_artifact=True,`
`141`	`142`	`)`
`142`	`143`	`### YOUR CODE ENDS HERE ###`
`143`	`144`