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_no_chunked_prefill.py
+++ b/test/srt/test_no_chunked_prefill.py
@@ -2,12 +2,13 @@ import unittest

 from sglang.test.test_utils import (
    DEFAULT_MODEL_NAME_FOR_TEST,
+    CustomTestCase,
    run_bench_serving,
    run_mmlu_test,
 )


-class TestNoChunkedPrefill(unittest.TestCase):
+class TestNoChunkedPrefill(CustomTestCase):

    def test_no_chunked_prefill(self):
        run_mmlu_test(