Remove max_concurrency from BOHB kwargs (#220)

ray-project · Sep 23, 2021 · dc71398 · dc71398
1 parent c7f2d75
commit dc71398
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tune_sklearn/tune_search.py b/tune_sklearn/tune_search.py
@@ -731,14 +731,8 @@ def _tune_run(self, config, resources_per_trial, tune_params=None):
             elif self.search_optimization == "bohb":
                 if override_search_space:
                     search_space = self._get_bohb_config_space()
-                if self.seed:
-                    warnings.warn("'seed' is not implemented for BOHB.")
-                if "max_concurrent" not in search_kwargs and isinstance(
-                        self.n_jobs, int) and self.n_jobs > 0:
-                    search_kwargs["max_concurrent"] = self.n_jobs
-                search_algo = TuneBOHB(space=search_space, **search_kwargs)
-                # search_algo = TuneBOHB(
-                #     space=search_space, seed=self.seed, **search_kwargs)
+                search_algo = TuneBOHB(
+                    space=search_space, seed=self.seed, **search_kwargs)
                 run_args["search_alg"] = search_algo
 
             elif self.search_optimization == "optuna":