TST Fixes test error

thomasjpfan · thomasjpfan · commit 4cc6c51419cc · 2022-02-25T20:47:00.000-05:00
diff --git a/sklearn/inspection/_plot/tests/test_boundary_decision_display.py b/sklearn/inspection/_plot/tests/test_boundary_decision_display.py
@@ -225,7 +225,7 @@ def fit(self, X, y):
 
 
 @pytest.mark.parametrize("response_method", ["auto", "predict", "predict_proba"])
-def test_multilabel_classifier_error(response_method):
+def test_multilabel_classifier_error(pyplot, response_method):
     """Check that multilabel classifier raises correct error."""
     X, y = make_multilabel_classification(random_state=0)
     X = X[:, :2]
@@ -241,7 +241,7 @@ def test_multilabel_classifier_error(response_method):
 
 
 @pytest.mark.parametrize("response_method", ["auto", "predict", "predict_proba"])
-def test_multi_output_multi_class_classifier_error(response_method):
+def test_multi_output_multi_class_classifier_error(pyplot, response_method):
     """Check that multi-output multi-class classifier raises correct error."""
     X = np.asarray([[0, 1], [1, 2]])
     y = np.asarray([["tree", "cat"], ["cat", "tree"]])
