From 53ea6ad8304f5d16bab97527ea1f4a19cfb25588 Mon Sep 17 00:00:00 2001 From: Lu Fang <30275821+houseroad@users.noreply.github.com> Date: Wed, 5 Mar 2025 13:41:18 -0800 Subject: [PATCH] [V1][Easy] Add empty allowed_token_ids in the v1 sampler test (#14308) Signed-off-by: Lu Fang --- tests/v1/sample/test_sampling_params_e2e.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tests/v1/sample/test_sampling_params_e2e.py b/tests/v1/sample/test_sampling_params_e2e.py index f17d4b77afc7f..4e88feae44dd6 100644 --- a/tests/v1/sample/test_sampling_params_e2e.py +++ b/tests/v1/sample/test_sampling_params_e2e.py @@ -113,6 +113,10 @@ def test_allowed_token_ids(model): PROMPT, SamplingParams(allowed_token_ids=allowed_token_ids)) assert output[0].outputs[0].token_ids[-1] == TOKEN_ID + # Reject empty allowed_token_ids. + with pytest.raises(ValueError): + _ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[])) + # Reject negative token id. with pytest.raises(ValueError): _ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[-1]))