unify objective function

Signed-off-by: helenxie-bit <[email protected]>
kubeflow · Jan 23, 2025 · b1d07ce · b1d07ce
1 parent 8cd3d0c
commit b1d07ce
Showing 1 changed file with 3 additions and 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
@@ -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,
     ),