Skip slow BalancedBatchSampler test

michaelbyun · Mar 22, 2023 · 03ba6e0 · 03ba6e0
1 parent bba24d8
commit 03ba6e0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/test_samplers.py b/tests/test_samplers.py
@@ -3,9 +3,10 @@
 from elk.extraction import BalancedBatchSampler, BalancedSampler
 from elk.utils import assert_type, infer_label_column
 from itertools import islice
-import numpy as np
+import pytest
 
 
+@pytest.mark.skip(reason="This test is too slow to run on every commit")
 def test_output_batches_are_balanced():
  # Load an example dataset for testing
  dataset = assert_type(