Skip to content

Commit d9c22b9

Browse files
committed
Format with yapf
1 parent 695a62c commit d9c22b9

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/v1/tpu/worker/test_tpu_model_runner.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -121,11 +121,11 @@ def _is_req_state_block_table_match(model_runner, req_id: str) -> bool:
121121
req_index = model_runner.input_batch.req_id_to_index[req_id]
122122
multi_group_block_table = model_runner.input_batch.block_table
123123
req_state = model_runner.requests[req_id]
124-
124+
125125
# Access the first block table from MultiGroupBlockTable
126126
# This is safe since we currently only use single KV cache groups
127127
block_table = multi_group_block_table[0]
128-
128+
129129
# req_state.block_ids is now list[list[int]] for MultiGroupBlockTable
130130
# Extract the first group's block IDs
131131
if isinstance(req_state.block_ids[0], list):
@@ -134,10 +134,10 @@ def _is_req_state_block_table_match(model_runner, req_id: str) -> bool:
134134
else:
135135
# Legacy format: list[int] - use directly
136136
req_block_ids = req_state.block_ids
137-
137+
138138
if block_table.num_blocks_per_row[req_index] != len(req_block_ids):
139139
return False
140-
140+
141141
num_blocks = block_table.num_blocks_per_row[req_index]
142142
block_table_values = block_table.block_table_np[req_index, :num_blocks]
143143
return (block_table_values == req_block_ids).all()

0 commit comments

Comments
 (0)