From c398ad513f4a6e6577af6d33f0601ae180862acc Mon Sep 17 00:00:00 2001 From: pengwa Date: Thu, 23 Jun 2022 23:23:36 +0800 Subject: [PATCH] Fix orttraining-linux-ci-pipeline - Symbolic shape infer (#11965) fix symbolic shape error due to upgraded numpy + legacy sympy --- tools/ci_build/build.py | 2 +- .../github/linux/docker/scripts/manylinux/requirements.txt | 2 +- tools/ci_build/github/linux/docker/scripts/requirements.txt | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py index 33db7190aeff..f7567fbee819 100644 --- 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 index a13110035b40..1353a36ea87b 100644 --- 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 index 50352e5adc86..8e57493478ab 100644 --- 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