Skip to content

Commit 62dd958

Browse files
authored
Remove retry in nightly tests (#4846)
1 parent 7203117 commit 62dd958

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

test/srt/test_nightly_gsm8k_eval.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -128,7 +128,8 @@ def check_model_scores(results):
128128
raise AssertionError("\n".join(failed_models))
129129

130130

131-
class TestNightlyGsm8KEval(CustomTestCase):
131+
# Do not use `CustomTestCase` since `test_mgsm_en_all_models` does not want retry
132+
class TestNightlyGsm8KEval(unittest.TestCase):
132133
@classmethod
133134
def setUpClass(cls):
134135
cls.model_groups = [

0 commit comments

Comments
 (0)