test added

Signed-off-by: jeongin601 <0200angela@gmail.com>
vllm-project · Nov 12, 2024 · e010834 · e010834
1 parent c0eabca
commit e010834
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/tests/spec_decode/test_batch_expansion.py b/tests/spec_decode/test_batch_expansion.py
@@ -90,10 +90,14 @@ def test_create_single_target_seq_group_metadata(k: int):
     )
 
     assert output.request_id == input_seq_group_metadata.request_id
-    assert output.sampling_params.repetition_penalty == input_seq_group_metadata.sampling_params.repetition_penalty
-    assert output.sampling_params.temperature == input_seq_group_metadata.sampling_params.temperature
-    assert output.sampling_params.top_p == input_seq_group_metadata.sampling_params.top_p
-    assert output.sampling_params.top_k == input_seq_group_metadata.sampling_params.top_k
+    assert output.sampling_params.repetition_penalty == \
+        input_seq_group_metadata.sampling_params.repetition_penalty
+    assert output.sampling_params.temperature == \
+        input_seq_group_metadata.sampling_params.temperature
+    assert output.sampling_params.top_p == \
+        input_seq_group_metadata.sampling_params.top_p
+    assert output.sampling_params.top_k == \
+        input_seq_group_metadata.sampling_params.top_k
     assert len(output.seq_data) == 1
     assert output.seq_data[target_seq_id].get_prompt_token_ids() == tuple(
         prompt_tokens)