Skip to content

Commit 2d2fc36

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

File tree

1 file changed

+11
-6
lines changed

1 file changed

+11
-6
lines changed

vllm_ascend/distributed/llmdatadist_c_mgr_connector.py

Lines changed: 11 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,11 @@ 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_msg[0]
377+
decode_tp_rank = decode_msg[1]
378+
decode_tp_size = decode_msg[2]
376379
with self.thread_lock:
377-
if self._increment_task_count(finished_req_id,
380+
if self._increment_task_count(finished_req_id,
378381
decode_tp_rank,
379382
decode_tp_size):
380383
logger.debug(
@@ -387,7 +390,7 @@ def listen_for_agent_metadata_req(self, event: threading.Event):
387390
f"LLMDataDistCMgrConnectorWorker: Receiving unexpected request event {event_msg} from remote !"
388391
)
389392

390-
def _increment_task_count(self, request_id: str, tp_rank: int,
393+
def _increment_task_count(self, request_id: str, tp_rank: int,
391394
decode_tp_size: int):
392395
if tp_rank in self.done_receiving_counts[request_id]:
393396
logger.warning(
@@ -752,8 +755,10 @@ def send_finish_to_remote(self, host: str, port: int, request_id):
752755
url = f"tcp://{host}:{port}"
753756
logger.debug(f"Sending finished to remote: {url}")
754757
msg_encoder = msgspec.msgpack.Encoder()
755-
msg_send = msg_encoder.encode(
756-
[LLMDataDistCMgrEvent.ReqForFinished, [request_id, self.tp_rank, self.tp_size]])
758+
msg_send = msg_encoder.encode([
759+
LLMDataDistCMgrEvent.ReqForFinished,
760+
[request_id, self.tp_rank, self.tp_size]
761+
])
757762
with zmq_ctx(zmq.REQ, url) as sock: # type: ignore[attr-defined]
758763
try:
759764
sock.send(msg_send)

0 commit comments

Comments
 (0)