add try except for trapezoid import

mtsokol · mtsokol · commit e522e0171a05 · 2023-08-30T19:59:36.000+02:00
diff --git a/sklearn/metrics/_plot/tests/test_precision_recall_display.py b/sklearn/metrics/_plot/tests/test_precision_recall_display.py
@@ -2,7 +2,12 @@
 
 import numpy as np
 import pytest
-from scipy.integrate import trapz as trapezoid
+
+try:
+    from scipy.integrate import trapezoid
+except ImportError:
+    # NOTE: remove once 1.6.0 is minimum supported scipy version
+    from scipy.integrate import trapz as trapezoid
 
 from sklearn.compose import make_column_transformer
 from sklearn.datasets import load_breast_cancer, make_classification
diff --git a/sklearn/metrics/_plot/tests/test_roc_curve_display.py b/sklearn/metrics/_plot/tests/test_roc_curve_display.py
@@ -1,7 +1,12 @@
 import numpy as np
 import pytest
 from numpy.testing import assert_allclose
-from scipy.integrate import trapz as trapezoid
+
+try:
+    from scipy.integrate import trapezoid
+except ImportError:
+    # NOTE: remove once 1.6.0 is minimum supported scipy version
+    from scipy.integrate import trapz as trapezoid
 
 from sklearn.compose import make_column_transformer
 from sklearn.datasets import load_breast_cancer, load_iris
diff --git a/sklearn/metrics/_ranking.py b/sklearn/metrics/_ranking.py
@@ -24,10 +24,15 @@
 from numbers import Integral, Real
 
 import numpy as np
-from scipy.integrate import trapz as trapezoid
 from scipy.sparse import csr_matrix, issparse
 from scipy.stats import rankdata
 
+try:
+    from scipy.integrate import trapezoid
+except ImportError:
+    # NOTE: remove once 1.6.0 is minimum supported scipy version
+    from scipy.integrate import trapz as trapezoid
+
 from ..exceptions import UndefinedMetricWarning
 from ..preprocessing import label_binarize
 from ..utils import (
