Skip to content

Commit 7d94b65

Browse files
committed
fix trainer test
1 parent 8e41cb6 commit 7d94b65

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/trainer/trainer_test.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1026,8 +1026,8 @@ def test_trainer(self):
10261026
with open(exp_save_path, "r", encoding="utf-8") as f:
10271027
lines = f.readlines()
10281028
self.assertTrue(
1029-
len(lines) > 2 * 4 * 2
1030-
) # total_steps * repeat_times * batch_size * min_waited_tasks
1029+
len(lines) >= 2 * 4 * 2
1030+
) # at least contain total_steps * repeat_times * batch_size * min_waited_tasks
10311031

10321032
def tearDown(self):
10331033
# remove dir only when the test passed

0 commit comments

Comments
 (0)