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_torch_tp.py
+++ b/test/srt/test_torch_tp.py
@@ -1,9 +1,9 @@
 import unittest

-from sglang.test.test_utils import is_in_ci, run_bench_one_batch
+from sglang.test.test_utils import CustomTestCase, is_in_ci, run_bench_one_batch


-class TestTorchTP(unittest.TestCase):
+class TestTorchTP(CustomTestCase):
    def test_torch_native_llama(self):
        output_throughput = run_bench_one_batch(
            "meta-llama/Meta-Llama-3-8B",