diff --git a/tests/generate/vllm_sampler_test.py b/tests/generate/vllm_sampler_test.py index 1893439d3..4de488fa2 100644 --- a/tests/generate/vllm_sampler_test.py +++ b/tests/generate/vllm_sampler_test.py @@ -283,7 +283,7 @@ def test_vllm_sampler_run_in_executor_concurrency(self): base_prompts[5]: ["winter"], base_prompts[6]: ["romeo", "juliet"], base_prompts[7]: ["pancake"], - base_prompts[8]: ["100", "celsius"], + base_prompts[8]: ["100", "212"], base_prompts[9]: ["seven", "eight"], } prompt_expectations = [