Slightly relax test_output_is_roughly_balanced

michaelbyun · Mar 22, 2023 · 15ab351 · 15ab351
1 parent 03ba6e0
commit 15ab351
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_samplers.py b/tests/test_samplers.py
@@ -43,7 +43,7 @@ def test_output_is_roughly_balanced():
 
  # Count the number of samples for each label
  counter = Counter()
- for sample in islice(reservoir, 2000):
+ for sample in islice(reservoir, 3000):
  counter[sample[col]] += 1
 
  # Check if the output is roughly balanced