Add retry for flaky tests in CI (#4755)

2025-03-26 07:53:12 +08:00
parent 52029bd1e3
commit 15ddd84322
112 changed files with 273 additions and 152 deletions
--- a/test/srt/test_abort.py
+++ b/test/srt/test_abort.py
@@ -5,10 +5,10 @@ from concurrent.futures import ThreadPoolExecutor

 import requests

-from sglang.test.test_utils import run_and_check_memory_leak
+from sglang.test.test_utils import CustomTestCase, run_and_check_memory_leak


-class TestAbort(unittest.TestCase):
+class TestAbort(CustomTestCase):
    def workload_func(self, base_url, model):
        def process_func():
            def run_one(_):