Lint

Author: sphish

deep_ep/buffer.py

@@ -35,7 +35,7 @@ def __init__(self, group: Optional[dist.ProcessGroup],
                  allow_nvlink_for_low_latency_mode: bool = True,
                  allow_mnnvl: bool = False,
                  explicitly_destroy: bool = False,
-                 comm: Optional["mpi4py.MPI.Comm"] = None) -> None:
+                 comm: Optional["mpi4py.MPI.Comm"] = None) -> None:  # noqa: F821
         """
         Initialize the communication buffer.
 

deep_ep/utils.py

@@ -1,11 +1,10 @@
 import os
-import subprocess
 import torch
 import torch.distributed as dist
 from typing import Any, Optional, Tuple
 
 # noinspection PyUnresolvedReferences
-from deep_ep_cpp import Config, EventHandle
+from deep_ep_cpp import EventHandle
 
 
 class EventOverlap: