Fix

allenai · Oct 23, 2024 · ffe470b · ffe470b
1 parent 180dde0
commit ffe470b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/test_dataprep.py b/tests/test_dataprep.py
@@ -151,10 +151,10 @@ def testListTargetAnchorLength(self):
 
         print(zero_count, full_count)
 
-        self.assertTrue(0.05 <= zero_ratio <= 0.15,
-                    f"Expected zero-length ratio around 0.10, got {zero_ratio:.2f}")
-        self.assertTrue(0.85 <= full_ratio <= 0.95,
-                    f"Expected full-length ratio around 0.90, got {full_ratio:.2f}")
+        self.assertTrue(0.45 <= zero_ratio <= 0.55,
+                    f"Expected zero-length ratio around 0.5, got {zero_ratio:.2f}")
+        self.assertTrue(0.45 <= full_ratio <= 0.55,
+                    f"Expected full-length ratio around 0.5, got {full_ratio:.2f}")
 
         # Verify total adds up to 100%
         self.assertEqual(zero_count + full_count, num_iterations,