From c810903fc955b48de96af5c451f6a12d0c06eeeb Mon Sep 17 00:00:00 2001 From: qli88 Date: Tue, 16 Dec 2025 17:20:46 +0000 Subject: [PATCH] ruff Signed-off-by: qli88 --- tests/samplers/test_no_bad_words.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/samplers/test_no_bad_words.py b/tests/samplers/test_no_bad_words.py index 9d58c8f1a613e..5721efcdeaf7e 100644 --- a/tests/samplers/test_no_bad_words.py +++ b/tests/samplers/test_no_bad_words.py @@ -96,7 +96,8 @@ class TestTwoTokenBadWord: def test_two_token_bad_word(self, vllm_runner): # TODO: Remove once graph mode is fixed for distilbert/distilgpt2 on ROCm. - eager_mode = current_platform.is_rocm() + + eager_mode = current_platform.is_rocm() with vllm_runner(self.MODEL, dtype="half", enforce_eager=eager_mode) as llm: output_token_ids = self._generate(llm)