Backwards compatibility with NumPy v1 (#835)

Author: czaloom

src/valor_lite/classification/computation.py

@@ -1,6 +1,8 @@
 import numpy as np
 from numpy.typing import NDArray
 
+import valor_lite.classification.numpy_compatibility as npc
+
 
 def _compute_rocauc(
     data: NDArray[np.float64],
@@ -56,7 +58,7 @@ def _compute_rocauc(
     np.maximum.accumulate(tpr, axis=1, out=tpr)
 
     # compute rocauc
-    rocauc = np.trapezoid(x=fpr, y=tpr, axis=1)
+    rocauc = npc.trapezoid(x=fpr, y=tpr, axis=1)
 
     # compute mean rocauc
     mean_rocauc = rocauc.mean()

src/valor_lite/classification/numpy_compatibility.py

@@ -0,0 +1,13 @@
+import numpy as np
+from numpy.typing import NDArray
+
+try:
+    _numpy_trapezoid = np.trapezoid  # numpy v2
+except AttributeError:
+    _numpy_trapezoid = np.trapz  # numpy v1
+
+
+def trapezoid(
+    x: NDArray[np.float64], y: NDArray[np.float64], axis: int
+) -> NDArray[np.float64]:
+    return _numpy_trapezoid(x=x, y=y, axis=axis)  # type: ignore - NumPy compatibility