fix

Signed-off-by: Your Name <[email protected]>
kubeflow · Jan 17, 2025 · beb5443 · beb5443
1 parent edfa973
commit beb5443
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/cmd/suggestion/optuna/v1beta1/requirements.txt b/cmd/suggestion/optuna/v1beta1/requirements.txt
@@ -3,4 +3,5 @@ protobuf>=4.21.12,<5
 googleapis-common-protos==1.53.0
 optuna==3.6.0
 torch==2.5.1
-scipy==1.15.1
+scipy==1.15.1
+cmaes==0.11.1
diff --git a/test/unit/v1beta1/suggestion/test_optuna_service.py b/test/unit/v1beta1/suggestion/test_optuna_service.py
@@ -120,7 +120,6 @@ def test_get_suggestion(self, algorithm_name, algorithm_settings):
         )
 
         response, metadata, code, details = get_suggestion.termination()
-        print(response)
         assert code == grpc.StatusCode.OK
         assert 2 == len(response.parameter_assignments)
 
@@ -486,7 +485,7 @@ def test_get_suggestion(self, algorithm_name, algorithm_settings):
                 grpc.StatusCode.INVALID_ARGUMENT,
             ],
             # [BAYESIANOPTIMIZATION] Invalid parameter name
-            # ["bayesianoptimization", {"invalid": "5"}, 100, [], grpc.StatusCode.INVALID_ARGUMENT],
+            ["bayesianoptimization", {"invalid": "5"}, 100, [], grpc.StatusCode.INVALID_ARGUMENT],
         ],
     )
     def test_validate_algorithm_settings(