From 760142fb1db60d76acf979fd243dab054ede2969 Mon Sep 17 00:00:00 2001 From: Baris Can Durak Date: Thu, 28 Nov 2024 14:41:03 +0100 Subject: [PATCH] fixing the call --- .../hp_tuning_single_search.py | 1 + 1 file changed, 1 insertion(+) diff --git a/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py b/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py index 0a5b022..261ab71 100644 --- a/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py +++ b/template/steps/{% if hyperparameters_tuning %}hp_tuning{% endif %}/hp_tuning_single_search.py @@ -82,6 +82,7 @@ def hp_tuning_single_search( log_metadata( metadata={"metric": float(score)}, artifact_name="hp_result", + infer_artifact=True, ) ### YOUR CODE ENDS HERE ### return cv.best_estimator_