FIX add tests to skip

neurodata · Dec 4, 2024 · 7071822 · 7071822
1 parent 2f475b4
commit 7071822
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/treeple/tree/tests/test_tree.py b/treeple/tree/tests/test_tree.py
@@ -227,12 +227,15 @@ def test_sklearn_compatible_estimator(estimator, check):
     # TODO: remove when we can replicate the CI error...
     if isinstance(
         estimator, (PatchObliqueDecisionTreeClassifier, ExtraObliqueDecisionTreeClassifier)
-    ) and check.func.__name__ in [
-        "check_fit_score_takes_y",
+    ) and check.func.__name__ in ["check_fit_score_takes_y"]:
+        pytest.skip()
+
+    if check.func.__name__ in [
         "check_sample_weight_equivalence_on_sparse_data",
         "check_sample_weight_equivalence_on_dense_data",
     ]:
         pytest.skip()
+
     check(estimator)