[serve] Fix long running failure test (ray-project#8863)

rkooo567 · Jun 11, 2020 · 07112dd · 07112dd
1 parent e6215d2
commit 07112dd
Showing 1 changed file with 3 additions and 5 deletions.
diff --git a/ci/long_running_tests/workloads/serve_failure.py b/ci/long_running_tests/workloads/serve_failure.py
@@ -6,7 +6,6 @@
 
 import ray
 from ray import serve
-from ray.serve.utils import retry_actor_failures
 from ray.cluster_utils import Cluster
 
 num_redis_shards = 1
@@ -42,11 +41,10 @@ def __init__(self, kill_period_s=1):
 
     def _get_all_serve_actors(self):
         master = serve.api._get_master_actor()
-        [router] = retry_actor_failures(master.get_router)
-        [http_proxy] = retry_actor_failures(master.get_http_proxy)
+        [router] = ray.get(master.get_router.remote())
+        [http_proxy] = ray.get(master.get_http_proxy.remote())
         all_handles = [master, router, http_proxy]
-        worker_handle_dict = retry_actor_failures(
-            master.get_all_worker_handles)
+        worker_handle_dict = ray.get(master.get_all_worker_handles.remote())
         for _, replica_dict in worker_handle_dict.items():
             all_handles.extend(list(replica_dict.values()))