Allow Numpy >=2.0

embeddings-benchmark · Sep 30, 2024 · febf99f · febf99f
1 parent 647c295
commit febf99f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mteb/evaluation/evaluators/utils.py b/mteb/evaluation/evaluators/utils.py
@@ -395,7 +395,7 @@ def abstention_curve(
         Returns:
             abst_curve: Abstention curve of length `len(abstention_rates)`
         """
-        conf_scores_argsort = np.argsort(conf_scores)
+        conf_scores_argsort = np.argsort(conf_scores, stable=False)
         abst_curve = np.zeros(len(abstention_rates))
 
         for i, rate in enumerate(abstention_rates):

diff --git a/pyproject.toml b/pyproject.toml
@@ -26,7 +26,7 @@ classifiers = [
 requires-python = ">=3.8"
 dependencies = [
     "datasets>=2.19.0",
-    "numpy>=1.0.0,<2.0.0", # note: https://github.com/huggingface/datasets/issues/6980
+    "numpy>=1.0.0,<3.0.0",
     "requests>=2.26.0",
     "scikit_learn>=1.0.2",
     "scipy>=0.0.0",