Change Numpy requirement to version == 2.3.

PiperOrigin-RevId: 771490711

Author: llcourage

.github/workflows/ci.yml

@@ -33,7 +33,7 @@ jobs:
     strategy:
       matrix:
         node-version: [18]
-        python-version: ["3.9", "3.10", "3.11"]
+        python-version: ["3.11"]
     defaults:
       run:
         shell: bash -l {0}

pyproject.toml

@@ -26,7 +26,7 @@ dependencies = [
   "Levenshtein>=0.21.1",
   "matplotlib>=3.7.1",
   "ml-collections>=0.1.1",
-  "numpy>=1.24.1,<2.0.0",
+  "numpy==2.3",
   "pandas[output-formatting]>=2.0.3",
   "Pillow>=10.0.0",
   "portpicker>=1.5.2",
@@ -36,7 +36,6 @@ dependencies = [
   "saliency>=0.1.3",
   "scikit-learn>=1.6.1",
   "scipy>=1.10.1",
-  "shap>=0.42.0,<0.46.0",
   "six>=1.16.0",
   "termcolor>=2.3.0",
   "tqdm>=4.64.0",
@@ -105,14 +104,12 @@ examples-generative-ai = [
 ]
 # LINT.ThenChange(./requirements_examples_generative_ai.txt)
 umap = [
-  "umap-learn==0.5.6"
 ]
 # LINT.IfChange
 test = [
   "lit-nlp[examples-discriminative-ai]",
   "lit-nlp[examples-generative-ai]",
   "lit-nlp[umap]",
-  "lime==0.2.0.1",
   "pytest>=7.4.0,<8.0.0",
   "webtest>=2.0",
 ]

requirements.txt

@@ -21,9 +21,9 @@ filelock>=3.12.3
 google-cloud-translate>=3.11.1
 ipython>=7.34.0
 Levenshtein>=0.21.1
-matplotlib>=3.6.0,<3.9.0
+matplotlib>=3.9.0
 ml-collections>=0.1.1
-numpy>=1.24.1,<2.0.0
+numpy==2.3
 pandas[output-formatting]>=2.0.3
 Pillow>=10.0.0
 portpicker>=1.5.2
@@ -33,7 +33,7 @@ sacrebleu>=2.3.1
 saliency>=0.1.3
 scikit-learn>=1.6.1
 scipy>=1.10.1
-shap>=0.42.0,<0.46.0
+shap>=0.46.0
 six>=1.16.0
 termcolor>=2.3.0
 tqdm>=4.64.0

requirements_test.txt

@@ -16,8 +16,6 @@
 -r requirements_examples_discriminative_ai.txt
 -r requirements_examples_generative_ai.txt
 
-lime==0.2.0.1
 pytest>=7.4.0,<8.0.0
-umap-learn==0.5.6
 webtest>=2.0
 # LINT.ThenChange(./pyproject.toml)