Rename bayesian-optimization to bayesianoptimization

kubeflow · Apr 23, 2020 · 45c71e2 · 45c71e2
1 parent 12313c1
commit 45c71e2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/suggestion/v1alpha3/chocolate/base_chocolate_service.py b/pkg/suggestion/v1alpha3/chocolate/base_chocolate_service.py
@@ -16,7 +16,7 @@
 DEPRECATED_ALGORITHM_NAME = {
     "chocolate-random": "random",
     "chocolate-quasirandom": "quasirandom",
-    "chocolate-bayesian-optimization": "bayesian-optimization",
+    "chocolate-bayesian-optimization": "bayesianoptimization",
     "chocolate-mocmaes": "mocmaes",
 }
 
@@ -76,7 +76,7 @@ def create_optimizer(self, algorithm_name):
         elif algorithm_name == "quasirandom":
             self.chocolate_optimizer = choco.QuasiRandom(
                 self.conn, chocolate_search_space, clear_db=True)
-        elif algorithm_name == "bayesian-optimization":
+        elif algorithm_name == "bayesianoptimization":
             self.chocolate_optimizer = choco.Bayes(
                 self.conn, chocolate_search_space, clear_db=True)
         # elif self.algorithm_name == "chocolate-CMAES":