From b1d07ce787bc59dde632ff6b38c92a4b5950843c Mon Sep 17 00:00:00 2001 From: helenxie-bit Date: Wed, 22 Jan 2025 22:39:11 -0800 Subject: [PATCH] unify objective function Signed-off-by: helenxie-bit --- sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py index 63969b4b790..0a78d75f3bb 100644 --- a/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py +++ b/sdk/python/v1beta1/kubeflow/katib/api/katib_client_test.py @@ -305,7 +305,7 @@ def create_experiment( "missing name", { "name": None, - "objective": lambda x: x, + "objective": lambda x: print(f"a={x}"), "parameters": {"a": katib.search.int(min=10, max=100)}, }, ValueError, @@ -314,7 +314,7 @@ def create_experiment( "invalid hybrid parameters - objective and model_provider_parameters", { "name": "tune_test", - "objective": lambda x: x, + "objective": lambda x: print(f"a={x}"), "model_provider_parameters": HuggingFaceModelParams( model_uri="hf://google-bert/bert-base-cased", transformer_type=transformers.AutoModelForSequenceClassification, @@ -334,7 +334,7 @@ def create_experiment( "missing parameters in custom objective tuning - lack parameters", { "name": "tune_test", - "objective": lambda x: x, + "objective": lambda x: print(f"a={x}"), }, ValueError, ),