Add suppress_health_check to test_stratified_split_dataset

nubank · May 23, 2024 · 8b3fe08 · 8b3fe08
1 parent d1cf236
commit 8b3fe08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/preprocessing/test_splitting.py b/tests/preprocessing/test_splitting.py
@@ -256,7 +256,7 @@ def assert_sample_size_per_class(data, target_column_name, expected_samples_per_
 @given(sample=gen_stratified_test_data(),
        random_state=st.integers(min_value=0, max_value=100),
        test_size=st.floats(min_value=0.2, max_value=0.8))
-@settings(suppress_health_check=HealthCheck.too_slow)
+@settings(suppress_health_check={HealthCheck.too_slow})
 def test_stratified_split_dataset(sample, random_state, test_size):
     expected_data, target_column_name, num_classes = sample