fix ci

Signed-off-by: linfeng-yuan <[email protected]>

Author: linfeng-yuan

vllm_ascend/core/recompute_scheduler.py

@@ -20,9 +20,11 @@
 
 import time
 from collections import defaultdict
-from dataclasses import dataclass
+from dataclasses import dataclass, fields
+from typing import Type, Union
 
 from vllm._bc_linter import bc_linter_include
+from vllm.config import SchedulerConfig, VllmConfig
 from vllm.distributed.ec_transfer.ec_connector.base import ECConnectorMetadata
 from vllm.distributed.kv_events import KVEventBatch
 from vllm.distributed.kv_transfer.kv_connector.v1.base import \
@@ -31,6 +33,7 @@
     KVConnectorStats
 from vllm.logger import init_logger
 from vllm.v1.core.kv_cache_manager import KVCacheBlocks
+from vllm.v1.core.sched.async_scheduler import AsyncScheduler
 from vllm.v1.core.sched.output import NewRequestData, SchedulerOutput
 from vllm.v1.core.sched.request_queue import (SchedulingPolicy,
                                               create_request_queue)
@@ -42,14 +45,10 @@
 from vllm.v1.request import Request, RequestStatus
 from vllm.v1.spec_decode.metrics import SpecDecodingStats
 from vllm.v1.utils import ConstantList, record_function_or_nullcontext
-from vllm.v1.core.sched.async_scheduler import AsyncScheduler
-from dataclasses import dataclass, fields
-from typing import Type, Union
-
-from vllm.config import SchedulerConfig, VllmConfig
 
 logger = init_logger(__name__)
 
+
 @dataclass
 class RecomputeSchedulerConfig(SchedulerConfig):
     scheduler_cls: Union[str, Type[object]] = (
@@ -64,8 +63,7 @@ def initialize_from_config(cls, vllm_config: VllmConfig):
         }
         if vllm_scheduler_config.async_scheduling:
             scheduler_config["scheduler_cls"] = (
-                "vllm_ascend.core.recompute_scheduler.AsyncRecomputeScheduler"
-            )
+                "vllm_ascend.core.recompute_scheduler.AsyncRecomputeScheduler")
         else:
             scheduler_config["scheduler_cls"] = (
                 "vllm_ascend.core.recompute_scheduler.RecomputeScheduler")
@@ -840,4 +838,4 @@ def update_from_output(
 class AsyncRecomputeScheduler(AsyncScheduler, RecomputeScheduler):
 
     def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
+        super().__init__(*args, **kwargs)