Limit the version of numpy (#25)

Signed-off-by: yuwenzho <[email protected]>
onnx · Jun 20, 2024 · bab644c · bab644c
1 parent 0f8c3a6
commit bab644c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
-numpy
+# onnxruntime doesn't support numpy>=2.0.0. The restriction will be removed once they fix it.
+numpy<2.0.0
 onnx
 onnxruntime
 onnxruntime-extensions

diff --git a/setup.py b/setup.py
@@ -54,7 +54,7 @@ def get_build_version():
             "onnxruntime",
             "onnxruntime-extensions",
             "psutil",
-            "numpy",
+            "numpy<2.0.0",
             "py-cpuinfo",
             "pydantic",
             "transformers",