From c4e46c973de6c606a315e13318c7f6e943e82e6d Mon Sep 17 00:00:00 2001
From: Shadi Noghabi <noghabi@google.com>
Date: Fri, 21 Aug 2026 21:15:27 -0700
Subject: [PATCH] fix vllm test breakage at head

PiperOrigin-RevId: 968854276
---
 tests/generate/vllm_sampler_test.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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 = [