[LINT] pre-commit lint

Signed-off-by: Csrayz <[email protected]>

Author: Csrayz

tests/ut/core/test_schedule_config.py

@@ -66,6 +66,7 @@ def test_initialize_from_config_with_override(self):
         self.assertEqual(ascend_config.encoder_cache_size, 2048)
         self.assertEqual(ascend_config.max_long_partial_prefills, 1)
         self.assertEqual(ascend_config.long_prefill_token_threshold, 512)
+
     def test_not_implemented_policy(self):
         with self.assertRaises(NotImplementedError) as context:
             AscendSchedulerConfig.initialize_from_config(

tests/ut/core/test_scheduler.py

@@ -268,7 +268,7 @@ def test_concurrent_partial_prefills_schedule(self):
         # Test initial scheduling
         output = scheduler.schedule()
         self.assertEqual(len(output.scheduled_new_reqs),
-            scheduler.scheduler_config.max_long_partial_prefills)
+                         scheduler.scheduler_config.max_long_partial_prefills)
         self.assertEqual(output.scheduled_cached_reqs.num_reqs, 0)
         self.assertEqual(len(output.finished_req_ids), 0)
 

vllm_ascend/core/schedule_config.py

@@ -16,12 +16,13 @@
 #
 
 from dataclasses import dataclass, fields
-from typing import Optional, Type, Union
+from typing import Type, Union
 
 from vllm.config import SchedulerConfig
 
 MAX_INT = 2147483647
 
+
 @dataclass
 class AscendSchedulerConfig(SchedulerConfig):
     enable_chunked_prefill: bool = False