Skip to content

Commit d036445

Browse files
committed
clean
Signed-off-by: junq <[email protected]>
1 parent 479d4c9 commit d036445

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unittest/_torch/executor/test_pytorch_model_engine.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -158,12 +158,12 @@ def test_pad_generation_requests(self) -> None:
158158
resource_manager = ResourceManager(
159159
{ResourceManagerType.KV_CACHE_MANAGER: kv_cache_manager})
160160

161-
seqlens_and_batch_sizes = [
161+
batch_sizes_and_seqlens = [
162162
(5, 1),
163163
(13, 1),
164164
(5, 25),
165165
]
166-
for (batch_size, max_seq_len) in seqlens_and_batch_sizes:
166+
for (batch_size, max_seq_len) in batch_sizes_and_seqlens:
167167
requests = [
168168
_create_request(max_seq_len, i) for i in range(batch_size)
169169
]

0 commit comments

Comments
 (0)