Skip to content

Commit 259ff3c

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

File tree

1 file changed

+9
-6
lines changed

1 file changed

+9
-6
lines changed

vllm_ascend/distributed/llmdatadist_c_mgr_connector.py

Lines changed: 9 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -338,7 +338,8 @@ def __init__(self, vllm_config: VllmConfig):
338338
self.finished_reqs: set[str] = set()
339339
self.soc_info = NPUSocInfo()
340340
# get decode tp size from extra config
341-
self.done_receiving_counts: defaultdict[str, set[int]] = defaultdict(set)
341+
self.done_receiving_counts: defaultdict[str,
342+
set[int]] = defaultdict(set)
342343

343344
def listen_for_agent_metadata_req(self, event: threading.Event):
344345
assert self.local_agent_metadata is not None
@@ -372,9 +373,9 @@ def listen_for_agent_metadata_req(self, event: threading.Event):
372373
f"LLMDataDistCMgrConnectorWorker: receiving unrecognized data {decode_msg}"
373374
)
374375
elif event_msg == LLMDataDistCMgrEvent.ReqForFinished:
375-
finished_req_id, decode_tp_rank, decode_tp_size = decode_msg[:3]
376+
finished_req_id, decode_tp_rank, decode_tp_size = decode_msg[0], decode_msg[1]. decode_msg[2]
376377
with self.thread_lock:
377-
if self._increment_task_count(finished_req_id,
378+
if self._increment_task_count(finished_req_id,
378379
decode_tp_rank,
379380
decode_tp_size):
380381
logger.debug(
@@ -387,7 +388,7 @@ def listen_for_agent_metadata_req(self, event: threading.Event):
387388
f"LLMDataDistCMgrConnectorWorker: Receiving unexpected request event {event_msg} from remote !"
388389
)
389390

390-
def _increment_task_count(self, request_id: str, tp_rank: int,
391+
def _increment_task_count(self, request_id: str, tp_rank: int,
391392
decode_tp_size: int):
392393
if tp_rank in self.done_receiving_counts[request_id]:
393394
logger.warning(
@@ -752,8 +753,10 @@ def send_finish_to_remote(self, host: str, port: int, request_id):
752753
url = f"tcp://{host}:{port}"
753754
logger.debug(f"Sending finished to remote: {url}")
754755
msg_encoder = msgspec.msgpack.Encoder()
755-
msg_send = msg_encoder.encode(
756-
[LLMDataDistCMgrEvent.ReqForFinished, [request_id, self.tp_rank, self.tp_size]])
756+
msg_send = msg_encoder.encode([
757+
LLMDataDistCMgrEvent.ReqForFinished,
758+
[request_id, self.tp_rank, self.tp_size]
759+
])
757760
with zmq_ctx(zmq.REQ, url) as sock: # type: ignore[attr-defined]
758761
try:
759762
sock.send(msg_send)

0 commit comments

Comments
 (0)