diff --git a/treeple/tree/tests/test_tree.py b/treeple/tree/tests/test_tree.py index e9cbe932..9ae39f4d 100644 --- 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)