try-this (#12338)

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>

try-this (#12338)
Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
7875b638 · Richard Liaw · GitHub · cf3c9198 · 7875b638
Unverified Commit 7875b638 authored Jun 24, 2021 by Richard Liaw Committed by GitHub Jun 24, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

setup.py setup.py +2 -2

No files found.
--- a/setup.py
+++ b/setup.py
@@ -125,7 +125,7 @@ _deps = [
    "pytest-sugar",
    "pytest-xdist",
    "python>=3.6.0",
-    "ray",
+    "ray[tune]",
    "recommonmark",
    "regex!=2019.12.17",
    "requests",
@@ -246,7 +246,7 @@ extras["sagemaker"] = deps_list("sagemaker")
 extras["deepspeed"] = deps_list("deepspeed")
 extras["fairscale"] = deps_list("fairscale")
 extras["optuna"] = deps_list("optuna")
-extras["ray"] = deps_list("ray")
+extras["ray"] = deps_list("ray[tune]")

 extras["integrations"] = extras["optuna"] + extras["ray"]