From 8b3fe08920e3c0abb8facbab0694b1aab0204d6b Mon Sep 17 00:00:00 2001 From: Pedro Spinosa Date: Wed, 22 May 2024 22:52:15 -0300 Subject: [PATCH] Add suppress_health_check to test_stratified_split_dataset --- tests/preprocessing/test_splitting.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/preprocessing/test_splitting.py b/tests/preprocessing/test_splitting.py index b820eb65..48dde394 100644 --- 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