remove nevergradsearch

erwulff · Feb 8, 2024 · b8e7f3a · b8e7f3a
1 parent e2ac610
commit b8e7f3a
Showing 1 changed file with 0 additions and 13 deletions.
diff --git a/mlpf/raytune/utils.py b/mlpf/raytune/utils.py
@@ -8,7 +8,6 @@
 from ray.tune.search.bayesopt import BayesOptSearch
 from ray.tune.search.bohb import TuneBOHB
 from ray.tune.search.hyperopt import HyperOptSearch
-from ray.tune.search.nevergrad import NevergradSearch
 from ray.tune.search.skopt import SkOptSearch
 
 # from ray.tune.search.hebo import HEBOSearch # HEBO is not yet supported
@@ -60,18 +59,6 @@ def get_raytune_search_alg(raytune_cfg, seeds=False):
             mode=raytune_cfg["default_mode"],
             convert_to_python=True,
         )
-    if raytune_cfg["search_alg"] == "nevergrad":
-        print("INFO: Using bayesian optimization from nevergrad")
-        import nevergrad as ng
-
-        return NevergradSearch(
-            optimizer=ng.optimizers.BayesOptim(
-                pca=False,
-                init_budget=raytune_cfg["nevergrad"]["n_random_steps"],
-            ),
-            metric=raytune_cfg["default_metric"],
-            mode=raytune_cfg["default_mode"],
-        )
     # HEBO is not yet supported
     # if (raytune_cfg["search_alg"] == "hebo") or (raytune_cfg["search_alg"] == "HEBO"):
     #     print("Using HEBOSearch")