Bigger array is more informative

allenai · Oct 30, 2024 · ffe7660 · ffe7660
1 parent e45d2c3
commit ffe7660
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/test/data/utils_test.py b/src/test/data/utils_test.py
@@ -14,9 +14,9 @@
 )
 
 
-@pytest.mark.limit_memory("11 KB")
+@pytest.mark.limit_memory("245 KB")
 def test_segment_documents_into_instances(tmp_path):
-    data = [1, 2, 3, 4, 0, 5, 6, 7, 8, 0, 1, 2, 3, 4, 5, 0] * 10
+    data = [1, 2, 3, 4, 0, 5, 6, 7, 8, 0, 1, 2, 3, 4, 5, 0] * 1000
     data_path = tmp_path / "data.npy"
     max_sequence_length = 4
     mmap = np.memmap(data_path, mode="w+", dtype=np.uint16, shape=(len(data),))