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_custom_allreduce.py
+++ b/test/srt/test_custom_allreduce.py
@@ -17,6 +17,7 @@ from sglang.srt.distributed.parallel_state import (
    graph_capture,
    initialize_model_parallel,
 )
+from sglang.test.test_utils import CustomTestCase


 def get_open_port() -> int:
@@ -54,7 +55,7 @@ def multi_process_parallel(
    ray.shutdown()


-class TestCustomAllReduce(unittest.TestCase):
+class TestCustomAllReduce(CustomTestCase):
    @classmethod
    def setUpClass(cls):
        random.seed(42)