Fix orttraining-linux-ci-pipeline - Symbolic shape infer (#11965)

fix symbolic shape error due to upgraded numpy + legacy sympy
microsoft · Jun 23, 2022 · c398ad5 · c398ad5
1 parent e24349b
commit c398ad5
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py
@@ -727,7 +727,7 @@ def is_docker():
 def install_python_deps(numpy_version=""):
     dep_packages = ["setuptools", "wheel", "pytest"]
     dep_packages.append("numpy=={}".format(numpy_version) if numpy_version else "numpy>=1.16.6")
-    dep_packages.append("sympy>=1.1")
+    dep_packages.append("sympy>=1.10")
     dep_packages.append("packaging")
     dep_packages.append("cerberus")
     run_subprocess([sys.executable, "-m", "pip", "install"] + dep_packages)

diff --git a/tools/ci_build/github/linux/docker/scripts/manylinux/requirements.txt b/tools/ci_build/github/linux/docker/scripts/manylinux/requirements.txt
@@ -5,5 +5,5 @@ setuptools>=41.4.0
 wheel
 onnx==1.12.0
 protobuf==3.18.1
-sympy==1.1.1
+sympy==1.10.1
 flatbuffers
diff --git a/tools/ci_build/github/linux/docker/scripts/requirements.txt b/tools/ci_build/github/linux/docker/scripts/requirements.txt
@@ -6,6 +6,6 @@ setuptools>=41.4.0
 wheel>=0.35.1
 onnx==1.12.0
 argparse
-sympy==1.1.1
+sympy==1.10.1
 flatbuffers
 protobuf==3.18.1