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/models/test_reward_models.py
+++ b/test/srt/models/test_reward_models.py
@@ -18,6 +18,7 @@ import unittest
 import torch

 from sglang.test.runners import HFRunner, SRTRunner
+from sglang.test.test_utils import CustomTestCase

 MODELS = [
    ("LxzGordon/URM-LLaMa-3.1-8B", 1, 4e-2),
@@ -41,7 +42,7 @@ CONVS = [
 ]


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

    @classmethod
    def setUpClass(cls):