From 7875b638cdabb3926ee5f0d34c760a22cc1c9c1d Mon Sep 17 00:00:00 2001
From: Richard Liaw <rliaw@berkeley.edu>
Date: Thu, 24 Jun 2021 01:13:17 -0700
Subject: [PATCH] try-this (#12338)

Signed-off-by: Richard Liaw <rliaw@berkeley.edu>
---
 setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 883f94268e..7b0eb6172b 100644
--- 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"]