We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 22c6e6d commit 2b91830Copy full SHA for 2b91830
tests/core/test_batch.py
@@ -445,7 +445,7 @@ def test_ConstantTokenNumSampler(self):
445
sample_count = 0
446
for batch_x, batch_y in data_iter:
447
sample_count += len(batch_x['seq_len'])
448
- self.assertTrue(sum(batch_x['seq_len'])<120)
+ self.assertTrue(sum(batch_x['seq_len'])<=120)
449
self.assertEqual(sample_count, num_samples)
450
451
"""
0 commit comments