Fixed AttributeError in metrics/roc_auc.py.

`scipy.integrate.trapezoid` replaced `scipy.integrate.trapz` in SciPy.
online-ml · Jul 7, 2024 · d6a01e5 · d6a01e5
1 parent acaf678
commit d6a01e5
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/river/metrics/roc_auc.py b/river/metrics/roc_auc.py
@@ -100,4 +100,10 @@ def safe_div(a, b):
             tprs[i] = safe_div(a=tp, b=tp + fn)
             fprs[i] = safe_div(a=fp, b=fp + tn)
 
-        return -integrate.trapz(x=fprs, y=tprs)
+        trapezoid = (
+            integrate.trapz  # For older/outdated versions of SciPy.
+            if hasattr(integrate, "trapz")
+            else integrate.trapezoid
+        )
+
+        return -trapezoid(x=fprs, y=tprs)