biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI/Build] Automatically retry flaky tests (#17856)

Browse Source 
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung
2025-05-09 23:55:17 +08:00
committed by GitHub
parent 200da9a517
commit 6e5595ca39
3 changed files with 2 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/tensorizer_loader/test_tensorizer.py
View File

@@ -28,7 +28,6 @@ from vllm.model_executor.model_loader.tensorizer import (TensorizerConfig,
from vllm.utils import PlaceholderModule, import_from_path
from ..utils import VLLM_PATH, RemoteOpenAIServer
from .conftest import retry_until_skip
try:
from tensorizer import EncryptionParams
@@ -325,7 +324,7 @@ def test_deserialized_encrypted_vllm_model_with_tp_has_same_outputs(
assert outputs == deserialized_outputs
@retry_until_skip(3)
@pytest.mark.flaky(reruns=3)
def test_vllm_tensorized_model_has_same_outputs(vllm_runner, tmp_path):
gc.collect()
torch.cuda.empty_cache()