Skip to content

GitLab

Unverified Commit 899136b8 authored by youkaichao's avatar youkaichao Committed by GitHub
Browse files

[ci] fix broken distributed-tests-4-gpus (#11937) 


Signed-off-by: default avataryoukaichao <youkaichao@gmail.com>
parent c9f09a4f
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tests/spec_decode/e2e/test_integration_dist_tp4.py
View file @ 899136b8
......@@ -108,7 +108,8 @@ def test_skip_speculation(common_llm_kwargs, per_test_common_llm_kwargs,
TODO: fix it to pass without raising Error. (#5814)
"""
with pytest.raises(openai.APIConnectionError):
with pytest.raises(
(openai.APIConnectionError, openai.InternalServerError)):
run_equality_correctness_test_tp(MAIN_MODEL,
common_llm_kwargs,
per_test_common_llm_kwargs,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment