Add retry for flaky tests in CI (#4755)

2025-03-25 16:53:12 -07:00
parent 52029bd1e3
commit 15ddd84322
112 changed files with 273 additions and 152 deletions
@@ -3,9 +3,10 @@ import unittest
 import torch

 from sglang.srt.utils import DynamicGradMode
+from sglang.test.test_utils import CustomTestCase


-class TestDynamicGradMode(unittest.TestCase):
+class TestDynamicGradMode(CustomTestCase):
    def test_inference(self):
        # Test inference_mode
        DynamicGradMode.set_inference_mode(True)