Add retry for flaky tests in CI (#4755)
This commit is contained in:
@@ -27,7 +27,7 @@ from sglang.test.runners import (
|
||||
check_close_model_outputs,
|
||||
get_dtype_str,
|
||||
)
|
||||
from sglang.test.test_utils import is_in_ci
|
||||
from sglang.test.test_utils import CustomTestCase, is_in_ci
|
||||
|
||||
_MAX_NEW_TOKENS = 8
|
||||
_PROMPTS = ["1+1=2, 1+2=3, 1+3=4, 1+4=5, 1+5=", "1*1=1, 1*2=2, 1*3=3, 1*4=4, 1*5="]
|
||||
@@ -73,7 +73,7 @@ ALL_OTHER_MODELS = [
|
||||
]
|
||||
|
||||
|
||||
class TestVerlEngine(unittest.TestCase):
|
||||
class TestVerlEngine(CustomTestCase):
|
||||
@classmethod
|
||||
def setUpClass(cls):
|
||||
multiprocessing.set_start_method("spawn")
|
||||
|
||||
Reference in New Issue
Block a user