f8d20d25bd4d3dd9bcfe4c332e833a400122a467,ci/long_running_tests/workloads/serve_failure.py,RandomKiller,_get_all_serve_actors,#RandomKiller#,43

Before Change

    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)
        all_handles = [master, router, http_proxy]
        worker_handle_dict = retry_actor_failures(
            master.get_all_worker_handles)

After Change

    def _get_all_serve_actors(self):
        master = serve.api._get_master_actor()
        [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 = ray.get(master.get_all_worker_handles.remote())
        for _, replica_dict in worker_handle_dict.items():
            all_handles.extend(list(replica_dict.values()))

        return all_handles

Instances