FIX add tests to skip

neurodata · Dec 3, 2024 · 0d036b0 · 0d036b0
1 parent f8c1452
commit 0d036b0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git 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)