From 0d036b0ce2633f2526239c7c850ac49d54ad3319 Mon Sep 17 00:00:00 2001 From: Haoyin Xu Date: Tue, 3 Dec 2024 14:23:37 -0500 Subject: [PATCH] FIX add tests to skip --- treeple/experimental/tests/test_sdf.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/treeple/experimental/tests/test_sdf.py b/treeple/experimental/tests/test_sdf.py index ba8326c6..e2d900fe 100644 --- a/treeple/experimental/tests/test_sdf.py +++ b/treeple/experimental/tests/test_sdf.py @@ -113,6 +113,10 @@ def test_max_samples(): def test_sklearn_compatible_estimator(estimator, check): # 1. check_class_weight_classifiers is not supported since it requires sample weight # XXX: can include this "generalization" in the future if it's useful - if check.func.__name__ in ["check_class_weight_classifiers", "check_sample_weight_equivalence"]: - pytest.skip() + if check.func.__name__ in [ + "check_class_weight_classifiers", + "check_sample_weight_equivalence", + "check_sample_weight_equivalence_on_dense_data", + "check_sample_weight_equivalence_on_sparse_data", + ]: check(estimator)