revert relax version

microsoft · Jun 26, 2024 · 1177530 · 1177530
1 parent 271ae76
commit 1177530
Show file tree

Hide file tree

Showing 9 changed files with 18 additions and 9 deletions.
diff --git a/tools/ci_build/github/linux/docker/inference/aarch64/python/cpu/scripts/requirements.txt b/tools/ci_build/github/linux/docker/inference/aarch64/python/cpu/scripts/requirements.txt
@@ -1,4 +1,5 @@
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 mypy
 pytest
 setuptools>=68.2.2

diff --git a/tools/ci_build/github/linux/docker/inference/x86_64/python/cpu/scripts/requirements.txt b/tools/ci_build/github/linux/docker/inference/x86_64/python/cpu/scripts/requirements.txt
@@ -1,4 +1,5 @@
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 mypy
 pytest
 setuptools>=68.2.2

diff --git a/tools/ci_build/github/linux/docker/inference/x86_64/python/cuda/scripts/requirements.txt b/tools/ci_build/github/linux/docker/inference/x86_64/python/cuda/scripts/requirements.txt
@@ -1,4 +1,5 @@
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 mypy
 pytest
 setuptools>=68.2.2

diff --git a/tools/ci_build/github/linux/docker/scripts/lort/requirements.txt b/tools/ci_build/github/linux/docker/scripts/lort/requirements.txt
@@ -8,7 +8,8 @@ onnx==1.16.1
 astunparse
 expecttest!=0.2.0
 hypothesis
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 psutil
 pyyaml
 requests

diff --git a/tools/ci_build/github/linux/docker/scripts/manylinux/requirements.txt b/tools/ci_build/github/linux/docker/scripts/manylinux/requirements.txt
@@ -1,4 +1,5 @@
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 mypy
 pytest
 setuptools>=68.2.2

diff --git a/.../github/linux/docker/scripts/training/ortmodule/stage1/requirements_rocm/requirements.txt b/.../github/linux/docker/scripts/training/ortmodule/stage1/requirements_rocm/requirements.txt
@@ -1 +1,2 @@
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
diff --git a/...ld/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt b/...ld/github/linux/docker/scripts/training/ortmodule/stage1/torch_eager_cpu/requirements.txt
@@ -5,6 +5,7 @@ setuptools>=68.2.2
 cerberus
 h5py
 scikit-learn
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 pandas
 parameterized
diff --git a/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt b/tools/ci_build/github/linux/docker/scripts/training/ortmodule/stage2/requirements.txt
@@ -1,6 +1,7 @@
 pandas
 scikit-learn
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 transformers==v4.36.0
 accelerate==0.25.0
 rsa==4.9

diff --git a/tools/ci_build/github/windows/eager/requirements.txt b/tools/ci_build/github/windows/eager/requirements.txt
@@ -1,6 +1,7 @@
 setuptools
 wheel
-numpy
+numpy==1.21.6 ; python_version < '3.9'
+numpy==2.0.0 ; python_version >= '3.9'
 typing_extensions
 torch==1.13.1
 parameterized