diff --git a/python/ray/tune/suggest/bayesopt.py b/python/ray/tune/suggest/bayesopt.py index cd0351d31..3bf1eedc1 100644 --- a/python/ray/tune/suggest/bayesopt.py +++ b/python/ray/tune/suggest/bayesopt.py @@ -70,7 +70,7 @@ class BayesOptSearch(Searcher): 'height': (-100, 100), } algo = BayesOptSearch(space, metric="mean_loss", mode="min") - tune.run(my_func, algo=algo) + tune.run(my_func, search_alg=algo) """ # bayes_opt.BayesianOptimization: Optimization object optimizer = None diff --git a/python/ray/tune/suggest/dragonfly.py b/python/ray/tune/suggest/dragonfly.py index 14b95b77e..e8de0c069 100644 --- a/python/ray/tune/suggest/dragonfly.py +++ b/python/ray/tune/suggest/dragonfly.py @@ -68,7 +68,7 @@ class DragonflySearch(Searcher): algo = DragonflySearch(optimizer, metric="objective", mode="max") - tune.run(my_func, algo=algo) + tune.run(my_func, search_alg=algo) Parameters: optimizer (dragonfly.opt.BlackboxOptimiser): Optimizer provided