fix format

vllm-project · Oct 28, 2024 · f1db8be · f1db8be
1 parent 8f31729
commit f1db8be
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/vllm/core/scheduler.py b/vllm/core/scheduler.py
@@ -1450,11 +1450,8 @@ def _append_slots(self,
             if len(cows) > 0:
                 blocks_to_copy.extend(cows)
 
-    def _preempt(
-        self,
-        seq_group: SequenceGroup,
-        blocks_to_swap_out: List[Tuple[int, int]]
-    ) -> PreemptionMode:
+    def _preempt(self, seq_group: SequenceGroup,
+                 blocks_to_swap_out: List[Tuple[int, int]]) -> PreemptionMode:
         # If preemption mode is not specified, we determine the mode as follows:
         # We use recomputation by default since it incurs lower overhead than
         # swapping. However, when the sequence group has multiple sequences