From 038a50af5296855a50764afd9f007cd552ff5c74 Mon Sep 17 00:00:00 2001 From: Richard Liaw Date: Sun, 20 Dec 2020 01:01:09 -0800 Subject: [PATCH] [tune] skopt fix-extra-import (#12970) Signed-off-by: Richard Liaw --- python/ray/tune/suggest/skopt.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/ray/tune/suggest/skopt.py b/python/ray/tune/suggest/skopt.py index c329b94d5..574be4f35 100644 --- a/python/ray/tune/suggest/skopt.py +++ b/python/ray/tune/suggest/skopt.py @@ -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'."