Skip to content

Commit 6ed4060

Browse files
committed
fix lint
Signed-off-by: underfituu <hzhucong@163.com>
1 parent a8a3954 commit 6ed4060

File tree

1 file changed

+7
-5
lines changed

1 file changed

+7
-5
lines changed

vllm_ascend/distributed/llmdatadist_c_mgr_connector.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -372,21 +372,23 @@ def listen_for_agent_metadata_req(self, event: threading.Event):
372372
f"LLMDataDistCMgrConnectorWorker: receiving unrecognized data {decode_msg}"
373373
)
374374
elif event_msg == LLMDataDistCMgrEvent.ReqForFinished:
375-
finished_req_id = decode_msg[0]
376-
decode_tp_rank = decode_msg[1]
377-
decode_tp_size = decode_msg[2]
375+
finished_req_id, decode_tp_rank, decode_tp_size = decode_msg[:3]
378376
with self.thread_lock:
379-
if self._increment_task_count(finished_req_id, decode_tp_rank, decode_tp_size):
377+
if self._increment_task_count(finished_req_id,
378+
decode_tp_rank,
379+
decode_tp_size):
380380
logger.debug(
381381
f"LLMDataDistCMgrConnectorWorker: Receiving request {finished_req_id} finished"
382382
)
383383
self.finished_reqs.add(finished_req_id)
384+
sock.send_multipart((identity, b"", msg_to_send))
384385
else:
385386
raise RuntimeError(
386387
f"LLMDataDistCMgrConnectorWorker: Receiving unexpected request event {event_msg} from remote !"
387388
)
388389

389-
def _increment_task_count(self, request_id: str, tp_rank: int, decode_tp_size: int):
390+
def _increment_task_count(self, request_id: str, tp_rank: int,
391+
decode_tp_size: int):
390392
if tp_rank in self.done_receiving_counts[request_id]:
391393
logger.warning(
392394
f"Received duplicate done signal for request {request_id} "

0 commit comments

Comments
 (0)