wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-04 09:36:46 +08:00
Code Issues Packages Projects Releases Wiki Activity

[tune] skopt fix-extra-import (#12970)

Browse Source 
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
This commit is contained in:
Richard Liaw
2020-12-20 01:01:09 -08:00
committed by GitHub
parent ec9ad4a56b
commit 038a50af52
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tune/suggest/skopt.py
+4 -4
View File
@@ -144,7 +144,7 @@ class SkOptSearch(Searcher):
"""
def __init__(self,
optimizer: Optional[sko.optimizer.Optimizer] = None,
optimizer: Optional["sko.optimizer.Optimizer"] = None,
space: Union[List[str], Dict[str, Union[Tuple, List]]] = None,
metric: Optional[str] = None,
mode: Optional[str] = None,
@@ -152,9 +152,9 @@ class SkOptSearch(Searcher):
evaluated_rewards: Optional[List] = None,
max_concurrent: Optional[int] = None,
use_early_stopped_trials: Optional[bool] = None):
assert sko is not None, """skopt must be installed!
You can install Skopt with the command:
`pip install scikit-optimize`."""
assert sko is not None, ("skopt must be installed! "
"You can install Skopt with the command: "
"`pip install scikit-optimize`.")
if mode:
assert mode in ["min", "max"], "`mode` must be 'min' or 'max'."