[CI][Test] Fix pooling truncation test after VLLMError hierarchy change (#50241)
Signed-off-by: Stefan Koncarevic <stefan.koncarevic@amd.com>
stefankoncarevic committed
Jul 29, 2026 at 14:15 UTC
625871b52c487077658582f72a0abbda559383e8
1 file changed
+3
-1
tests/models/language/pooling/test_truncation_control.py
+3
-1
@@ -2,6 +2,8 @@
2
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
3
import pytest
4
5
+from vllm.exceptions import VLLMValidationError
6
+
7
MODEL_NAME = "sentence-transformers/all-MiniLM-L12-v2"
8
max_model_len = 128
9
@@ -60,7 +62,7 @@ def test_bigger_truncation_size(
62
truncate_prompt_tokens = max_model_len + 1
63
64
with (
63
- pytest.raises(ValueError),
65
+ pytest.raises(VLLMValidationError),
66
vllm_runner(
67
model_name, runner="pooling", max_model_len=max_model_len
68
) as vllm_model,