From 2b91830c2617e829270b59c0f1ac1aff53a943d6 Mon Sep 17 00:00:00 2001
From: sleepy-owl <61391995+sleepy-owl@users.noreply.github.com>
Date: Tue, 23 Feb 2021 05:04:05 -0600
Subject: [PATCH] Flaky test fix (#352)

---
 tests/core/test_batch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/core/test_batch.py b/tests/core/test_batch.py
index 7bdcbda7..1e218800 100644
--- a/tests/core/test_batch.py
+++ b/tests/core/test_batch.py
@@ -445,7 +445,7 @@ class TestCase1(unittest.TestCase):
         sample_count = 0
         for batch_x, batch_y in data_iter:
             sample_count += len(batch_x['seq_len'])
-            self.assertTrue(sum(batch_x['seq_len'])<120)
+            self.assertTrue(sum(batch_x['seq_len'])<=120)
         self.assertEqual(sample_count, num_samples)
 
     """