wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-06 01:24:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Edward Oakes
2020-06-09 12:13:20 -05:00
committed by GitHub
parent 3d473600a8
commit f8d20d25bd
1 changed files with 3 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ci/long_running_tests/workloads/serve_failure.py
+3 -5
View File
@@ -6,7 +6,6 @@ import requests
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 @@ class RandomKiller:
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()))