restore

vllm-project · Sep 20, 2024 · f11d549 · f11d549
1 parent 9a6d06c
commit f11d549
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/samplers/test_beam_search.py b/tests/samplers/test_beam_search.py
@@ -27,7 +27,7 @@ def test_beam_search_single_input(
     max_tokens: int,
     beam_width: int,
 ) -> None:
-    example_prompts = example_prompts[1:2]
+    example_prompts = example_prompts[:1]
     with hf_runner(model, dtype=dtype) as hf_model:
         hf_outputs = hf_model.generate_beam_search(example_prompts, beam_width,
                                                    max_tokens)