Skip to content

Commit 09465dd

Browse files
youkaichaorasmith
authored andcommitted
[ci] fix broken distributed-tests-4-gpus (vllm-project#11937)
Signed-off-by: youkaichao <youkaichao@gmail.com>
1 parent 64cb56a commit 09465dd

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

tests/spec_decode/e2e/test_integration_dist_tp4.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -108,7 +108,8 @@ def test_skip_speculation(common_llm_kwargs, per_test_common_llm_kwargs,
108108
109109
TODO: fix it to pass without raising Error. (#5814)
110110
"""
111-
with pytest.raises(openai.APIConnectionError):
111+
with pytest.raises(
112+
(openai.APIConnectionError, openai.InternalServerError)):
112113
run_equality_correctness_test_tp(MAIN_MODEL,
113114
common_llm_kwargs,
114115
per_test_common_llm_kwargs,

0 commit comments

Comments
 (0)