run precommit

Author: zhtshr

lightx2v/disagg/README.md

@@ -140,4 +140,4 @@ bash scripts/disagg/run_dynamic.sh
 
 1. 多机运行时，`DISAGG_CONTROLLER_CFG` 里的 `bootstrap_addr`、`static_instance_slots` 和各 slot 的 `env` 会直接影响远端实例如何绑定网络与 Mooncake 地址。
 2. 如果遇到端口占用，优先检查 `scripts/disagg/kill_service.sh` 是否已经把旧实例和 proxy 清理干净。
-3. 如果需要了解 controller 配置文件本身的字段含义，可以继续查看 `configs/disagg/` 下对应 JSON。
+3. 如果需要了解 controller 配置文件本身的字段含义，可以继续查看 `configs/disagg/` 下对应 JSON。

lightx2v/disagg/rdma_client.py

@@ -7,21 +7,21 @@
 import time
 
 from lightx2v.disagg.rdma_base import (
-    AccessFlag,
-    AHAttr,
     CQ,
     GID,
-    GlobalRoute,
-    IBDevice,
     MR,
     PD,
     QP,
+    SGE,
+    WR,
+    AHAttr,
+    AccessFlag,
+    GlobalRoute,
+    IBDevice,
     QPAttr,
     QPCap,
     QPInitAttr,
     QPType,
-    SGE,
-    WR,
     WROpcode,
     e,
     get_device_list,

lightx2v/disagg/rdma_server.py

@@ -4,15 +4,15 @@
 import threading
 
 from lightx2v.disagg.rdma_base import (
-    AccessFlag,
-    AHAttr,
     CQ,
     GID,
-    GlobalRoute,
-    IBDevice,
     MR,
     PD,
     QP,
+    AHAttr,
+    AccessFlag,
+    GlobalRoute,
+    IBDevice,
     QPAttr,
     QPCap,
     QPInitAttr,

lightx2v/disagg/rdma_utils.py

@@ -7,7 +7,6 @@
 import socket
 import time
 
-
 logger = logging.getLogger(__name__)
 
 
@@ -232,4 +231,4 @@ def rtr_path_mtu_negotiated(ctx, port_num: int, peer_active_mtu: int | None) ->
         peer = int(peer_active_mtu)
     except (TypeError, ValueError):
         return local
-    return min(local, peer)
+    return min(local, peer)

lightx2v/disagg/services/base.py

@@ -1,9 +1,8 @@
-from abc import ABC
 import sys
+from abc import ABC
 
 from loguru import logger as loguru_logger
 
-
 loguru_logger.remove()
 loguru_logger.add(
     sys.stderr,

lightx2v/disagg/services/controller.py

@@ -1,15 +1,15 @@
 import ipaddress
 import json
-from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
 import os
 import shlex
-import signal
 import shutil
+import signal
 import socket
 import subprocess
 import sys
 import time
 from collections.abc import Mapping
+from http.server import BaseHTTPRequestHandler, ThreadingHTTPServer
 from pathlib import Path
 from threading import Event, Lock, Thread
 from typing import Any
@@ -501,11 +501,7 @@ def _with_env(base_cmd: str, env_map: dict[str, str]) -> str:
             return f"env {env_prefix} {base_cmd}"
 
         sidecar_cmd = _with_env(
-            (
-                f"{shlex.quote(python_executable)} "
-                "-m lightx2v.disagg.services.data_mgr_sidecar "
-                f"--push-addr {shlex.quote(push_addr)} --req-addr {shlex.quote(req_addr)}"
-            ),
+            (f"{shlex.quote(python_executable)} -m lightx2v.disagg.services.data_mgr_sidecar --push-addr {shlex.quote(push_addr)} --req-addr {shlex.quote(req_addr)}"),
             sidecar_env_vars,
         )
         cmd_with_python = [python_executable, *cmd[1:]]
@@ -547,9 +543,7 @@ def _with_env(base_cmd: str, env_map: dict[str, str]) -> str:
                     service_pid = None
 
         if sidecar_pid is None or service_pid is None:
-            raise RuntimeError(
-                f"failed to parse remote pids for {instance_type} rank={engine_rank} host={host}: stdout={completed.stdout!r} stderr={completed.stderr!r}"
-            )
+            raise RuntimeError(f"failed to parse remote pids for {instance_type} rank={engine_rank} host={host}: stdout={completed.stdout!r} stderr={completed.stderr!r}")
 
         sidecar_meta = {
             "mode": "remote",

lightx2v/disagg/services/decoder.py

@@ -143,7 +143,7 @@ def init(self, config):
         if data_bootstrap_addr is None or data_bootstrap_room is None:
             return
 
-        if not str(os.getenv("IS_CENTRALIZED", "0")).strip().lower() in {"1", "true", "yes", "on"}:
+        if str(os.getenv("IS_CENTRALIZED", "0")).strip().lower() not in {"1", "true", "yes", "on"}:
             try:
                 self._ensure_phase2_request_buffer()
             except Exception:

lightx2v/disagg/services/encoder.py

@@ -10,7 +10,6 @@
 
 import numpy as np
 import torch
-import zmq
 
 from lightx2v.disagg.conn import MONITOR_POLLING_PORT, REQUEST_POLLING_PORT, DataArgs, DataManager, DataPoll, DataSender, DisaggregationMode, DisaggregationPhase, ReqManager
 from lightx2v.disagg.monitor import Reporter

lightx2v/disagg/services/transformer.py

@@ -6,12 +6,11 @@
 from collections import deque
 from multiprocessing import resource_tracker, shared_memory
 from typing import Any, List, Optional
+from urllib.error import URLError
+from urllib.request import Request, urlopen
 
 import numpy as np
 import torch
-import zmq
-from urllib.error import URLError
-from urllib.request import Request, urlopen
 
 from lightx2v.disagg.conn import MONITOR_POLLING_PORT, REQUEST_POLLING_PORT, DataArgs, DataManager, DataPoll, DataReceiver, DataSender, DisaggregationMode, DisaggregationPhase, ReqManager
 from lightx2v.disagg.monitor import Reporter

lightx2v/disagg/utils.py

@@ -374,6 +374,7 @@ def load_wan_transformer(config: Dict[str, Any]):
     elif config.get("model_cls") == "wan2.2_moe":
         print("Loading MultiModelStruct module start", flush=True)
         from lightx2v.models.runners.wan.wan_runner import MultiModelStruct
+
         print("Loading MultiModelStruct module done", flush=True)
 
         high_noise_model_path = os.path.join(config["model_path"], "high_noise_model")