From 338428a1c3b443ecdc335b0cd775efa032c4e716 Mon Sep 17 00:00:00 2001 From: Adam Li Date: Wed, 16 Aug 2023 13:52:13 -0400 Subject: [PATCH] Fixed Signed-off-by: Adam Li --- sktree/tree/tests/test_tree.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sktree/tree/tests/test_tree.py b/sktree/tree/tests/test_tree.py index ce3d18304..46f39d9f0 100644 --- a/sktree/tree/tests/test_tree.py +++ b/sktree/tree/tests/test_tree.py @@ -225,7 +225,7 @@ def test_pickle_splitters(): def test_sklearn_compatible_estimator(estimator, check): # TODO: remove when we can replicate the CI error... if isinstance( - estimator, [PatchObliqueDecisionTreeClassifier, ExtraObliqueDecisionTreeClassifier] + estimator, (PatchObliqueDecisionTreeClassifier, ExtraObliqueDecisionTreeClassifier) ) and check.func.__name__ in ["check_fit_score_takes_y"]: pytest.skip() check(estimator)