From 4ee3c0e85b6b6b85089ab9b21c10167db95f47a8 Mon Sep 17 00:00:00 2001 From: Aleksandr Semikin Date: Fri, 12 Dec 2025 19:39:37 +0300 Subject: [PATCH] fixed_server_shuffling --- verl/experimental/agent_loop/agent_loop.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/verl/experimental/agent_loop/agent_loop.py b/verl/experimental/agent_loop/agent_loop.py index d2ac253cd4f..cc1d7437f04 100644 --- a/verl/experimental/agent_loop/agent_loop.py +++ b/verl/experimental/agent_loop/agent_loop.py @@ -70,7 +70,7 @@ def __init__(self, config: DictConfig, server_handles: list[ray.actor.ActorHandl random.shuffle(self.server_handles) # Least requests load balancing - self.weighted_serveres = [[0, (hash(server), server)] for server in server_handles] + self.weighted_serveres = [[0, idx, server] for idx, server in enumerate(self.server_handles)] heapq.heapify(self.weighted_serveres) # LRU cache to map request_id to server @@ -81,7 +81,7 @@ def _choose_server(self, request_id: str) -> ray.actor.ActorHandle: if request_id in self.request_id_to_server: return self.request_id_to_server[request_id] - server = self.weighted_serveres[0][1][1] + _, _, server = self.weighted_serveres[0] self.weighted_serveres[0][0] += 1 heapq.heapreplace(self.weighted_serveres, self.weighted_serveres[0]) self.request_id_to_server[request_id] = server