fix import sorting issue

Signed-off-by: ZihuiQian <qianzihui@huawei.com>

Author: ZihuiQian

examples/offline_disaggregated_prefill_npu.py

@@ -16,11 +16,10 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
+import multiprocessing as mp
 import os
 import time
-
 from multiprocessing import Event, Process
-import multiprocessing as mp
 
 
 def clean_up():

vllm_ascend/distributed/__init__.py

@@ -1,4 +1,5 @@
-from vllm.distributed.kv_transfer.kv_connector.factory import KVConnectorFactory
+from vllm.distributed.kv_transfer.kv_connector.factory import \
+    KVConnectorFactory
 
 KVConnectorFactory.register_connector(
     "AscendHcclConnector", "vllm_ascend.distributed.llmdatadist_connector",

vllm_ascend/distributed/llmdatadist_connector.py

@@ -14,15 +14,14 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 #
-import re
 import os
+import re
 import subprocess
 from typing import TYPE_CHECKING, List, Tuple, Union
 
 import torch
-import torchair
 import torch_npu
-
+import torchair
 from vllm.config import VllmConfig
 from vllm.distributed.kv_transfer.kv_connector.base import KVConnectorBase
 from vllm.logger import init_logger
@@ -31,15 +30,8 @@
 if TYPE_CHECKING:
     from vllm.worker.model_runner import ModelInputForGPUWithSamplingMetadata
 
-from llm_datadist import (
-    CacheDesc,
-    CacheKey,
-    CacheKeyByIdAndIndex,
-    DataType,
-    LLMClusterInfo,
-    LLMDataDist,
-    LLMRole,
-)
+from llm_datadist import (CacheDesc, CacheKey, CacheKeyByIdAndIndex, DataType,
+                          LLMClusterInfo, LLMDataDist, LLMRole)
 
 logger = init_logger(__name__)