Skip to content

Commit 2b91830

Browse files
authored
Flaky test fix (#352)
1 parent 22c6e6d commit 2b91830

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/core/test_batch.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -445,7 +445,7 @@ def test_ConstantTokenNumSampler(self):
445445
sample_count = 0
446446
for batch_x, batch_y in data_iter:
447447
sample_count += len(batch_x['seq_len'])
448-
self.assertTrue(sum(batch_x['seq_len'])<120)
448+
self.assertTrue(sum(batch_x['seq_len'])<=120)
449449
self.assertEqual(sample_count, num_samples)
450450

451451
"""

0 commit comments

Comments
 (0)