[hotfix] Lint formatting for new Tune optimizer ZOOpt (#8040)

* formatting

* removedill

* lint
This commit is contained in:
Richard Liaw
2020-04-16 09:24:30 -07:00
committed by GitHub
parent d4cae5f632
commit 4d8bf5635d
6 changed files with 21 additions and 12 deletions
+11 -6
View File
@@ -1,8 +1,12 @@
import copy
import logging
import dill as pickle
from zoopt import Dimension2, Parameter
from zoopt.algos.opt_algorithms.racos.sracos import SRacosTune
import ray.cloudpickle as pickle
try:
import zoopt
except ImportError:
zoopt = None
from ray.tune.suggest.suggestion import SuggestionAlgorithm
logger = logging.getLogger(__name__)
@@ -76,7 +80,7 @@ class ZOOptSearch(SuggestionAlgorithm):
metric="episode_reward_mean",
mode="min",
**kwargs):
assert zoopt is not None, "Zoopt not found - please install zoopt."
assert budget is not None, "`budget` should not be None!"
assert dim_dict is not None, "`dim_list` should not be None!"
assert type(max_concurrent) is int and max_concurrent > 0
@@ -99,9 +103,10 @@ class ZOOptSearch(SuggestionAlgorithm):
self._dim_keys.append(k)
_dim_list.append(dim_dict[k])
dim = Dimension2(_dim_list)
par = Parameter(budget=budget)
dim = zoopt.Dimension2(_dim_list)
par = zoopt.Parameter(budget=budget)
if _algo == "sracos" or _algo == "asracos":
from zoopt.algos.opt_algorithms.racos.sracos import SRacosTune
self.optimizer = SRacosTune(dimension=dim, parameter=par)
self.solution_dict = {}