From a454c815f1b350075db748e1302037ec11ba3961 Mon Sep 17 00:00:00 2001
From: Philipp Moritz <pcmoritz@gmail.com>
Date: Thu, 5 Dec 2019 18:29:41 -0800
Subject: [PATCH] Fix long running stress tests (#6374)

---
 ci/long_running_tests/workloads/actor_deaths.py     | 2 +-
 ci/long_running_tests/workloads/apex.py             | 2 +-
 ci/long_running_tests/workloads/impala.py           | 2 +-
 ci/long_running_tests/workloads/many_actor_tasks.py | 2 +-
 ci/long_running_tests/workloads/many_drivers.py     | 2 +-
 ci/long_running_tests/workloads/many_tasks.py       | 2 +-
 ci/long_running_tests/workloads/node_failures.py    | 2 +-
 ci/long_running_tests/workloads/pbt.py              | 2 +-
 8 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/ci/long_running_tests/workloads/actor_deaths.py b/ci/long_running_tests/workloads/actor_deaths.py
index 993a495f3..795960b93 100644
--- a/ci/long_running_tests/workloads/actor_deaths.py
+++ b/ci/long_running_tests/workloads/actor_deaths.py
@@ -9,7 +9,7 @@ import sys
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 1
 redis_max_memory = 10**8
diff --git a/ci/long_running_tests/workloads/apex.py b/ci/long_running_tests/workloads/apex.py
index 3b99dce6b..6b306f68e 100644
--- a/ci/long_running_tests/workloads/apex.py
+++ b/ci/long_running_tests/workloads/apex.py
@@ -5,7 +5,7 @@ from __future__ import division
 from __future__ import print_function
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 from ray.tune import run_experiments
 
 num_redis_shards = 5
diff --git a/ci/long_running_tests/workloads/impala.py b/ci/long_running_tests/workloads/impala.py
index 9fc9dadf6..56f1ce4d0 100644
--- a/ci/long_running_tests/workloads/impala.py
+++ b/ci/long_running_tests/workloads/impala.py
@@ -6,7 +6,7 @@ from __future__ import print_function
 
 import ray
 from ray.tune import run_experiments
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 5
 redis_max_memory = 10**8
diff --git a/ci/long_running_tests/workloads/many_actor_tasks.py b/ci/long_running_tests/workloads/many_actor_tasks.py
index e1a5de9ed..04c49f351 100644
--- a/ci/long_running_tests/workloads/many_actor_tasks.py
+++ b/ci/long_running_tests/workloads/many_actor_tasks.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 5
 redis_max_memory = 10**8
diff --git a/ci/long_running_tests/workloads/many_drivers.py b/ci/long_running_tests/workloads/many_drivers.py
index 7867f8124..2e540e046 100644
--- a/ci/long_running_tests/workloads/many_drivers.py
+++ b/ci/long_running_tests/workloads/many_drivers.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 from ray.test_utils import run_string_as_driver
 
 num_redis_shards = 5
diff --git a/ci/long_running_tests/workloads/many_tasks.py b/ci/long_running_tests/workloads/many_tasks.py
index 6aca51b49..c46826835 100644
--- a/ci/long_running_tests/workloads/many_tasks.py
+++ b/ci/long_running_tests/workloads/many_tasks.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 5
 redis_max_memory = 10**8
diff --git a/ci/long_running_tests/workloads/node_failures.py b/ci/long_running_tests/workloads/node_failures.py
index 693c47bac..0770e65ff 100644
--- a/ci/long_running_tests/workloads/node_failures.py
+++ b/ci/long_running_tests/workloads/node_failures.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 import time
 
 import ray
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 5
 redis_max_memory = 10**8
diff --git a/ci/long_running_tests/workloads/pbt.py b/ci/long_running_tests/workloads/pbt.py
index 936acab56..05298f198 100644
--- a/ci/long_running_tests/workloads/pbt.py
+++ b/ci/long_running_tests/workloads/pbt.py
@@ -7,7 +7,7 @@ from __future__ import print_function
 import ray
 from ray.tune import run_experiments
 from ray.tune.schedulers import PopulationBasedTraining
-from ray.tests.cluster_utils import Cluster
+from ray.cluster_utils import Cluster
 
 num_redis_shards = 5
 redis_max_memory = 10**8