yapf reformat vllm/core

vllm/block.py

@@ -5,6 +5,7 @@
 
 DEFAULT_LAST_ACCESSED_TIME: float = -1
 
+
 class PhysicalTokenBlock:
     """Represents the state of a block in the KV cache."""
 
@@ -59,8 +60,8 @@ def __getitem__(self, key):
         return self._blocks[key]
 
     def __iter__(self) -> Iterator[PhysicalTokenBlock]:
-      for i in range(len(self)):
-        yield self[i]
+        for i in range(len(self)):
+            yield self[i]
 
     def __setitem__(self, key, value):
         if isinstance(key, slice):

vllm/core/block_manager_v1.py

@@ -321,7 +321,8 @@ def _allocate_sequence(self, \
             for logical_idx in range(num_prompt_blocks):
                 if (self.block_sliding_window is not None
                         and logical_idx >= self.block_sliding_window):
-                    block = block_table[logical_idx % self.block_sliding_window]
+                    block = block_table[logical_idx %
+                                        self.block_sliding_window]
                     # Set the reference counts of the token blocks.
                     block.ref_count = ref_count
                 elif not is_encoder_decoder and self.enable_caching:

vllm/core/scheduler.py

@@ -288,7 +288,8 @@ def scheduler_running_outputs_builder():
 
 
 def scheduled_seq_group_builder():
-    return ScheduledSequenceGroup(SequenceGroup("", [], -1), token_chunk_size=0)
+    return ScheduledSequenceGroup(SequenceGroup("", [], -1),
+                                  token_chunk_size=0)
     # return ScheduledSequenceGroup(seq_group=None, token_chunk_size=0)