Update protobuf version in ci (#1951)

* update protobuf version in ci Signed-off-by: Deyu Huang <deyhuang@microsoft.com> * add change in unit_test Signed-off-by: Deyu Huang <deyhuang@microsoft.com>
onnx · May 27, 2022 · 6f5a673 · 6f5a673
1 parent aaab800
commit 6f5a673
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/ci_build/azure_pipelines/templates/keras2onnx_application_tests.yml b/ci_build/azure_pipelines/templates/keras2onnx_application_tests.yml
@@ -19,6 +19,8 @@ steps:
       python -m pip install --upgrade pip
       conda config --set always_yes yes --set changeps1 no
       pip install $(ONNX_PATH)
+      pip uninstall -y protobuf
+      pip install "protobuf<4.21.0"
       pip install h5py==2.9.0
       pip install parameterized
       $(INSTALL_TENSORFLOW)
@@ -81,7 +83,7 @@ steps:
       echo Test numpy installation... && python -c "import numpy"
       pip install %ONNX_PATH%
       pip uninstall -y protobuf
-      pip install protobuf
+      pip install "protobuf<4.21.0"
       pip install h5py==2.9.0
       pip install parameterized
       %INSTALL_TENSORFLOW%

diff --git a/ci_build/azure_pipelines/templates/keras2onnx_unit_test.yml b/ci_build/azure_pipelines/templates/keras2onnx_unit_test.yml
@@ -19,6 +19,8 @@ steps:
       python -m pip install --upgrade pip
       conda config --set always_yes yes --set changeps1 no
       pip install $(ONNX_PATH)
+      pip uninstall -y protobuf
+      pip install "protobuf<4.21.0"
       pip install h5py==2.9.0
       pip install parameterized
       pip install $(TENSORFLOW_PATH)
@@ -71,7 +73,7 @@ steps:
       echo Test numpy installation... && python -c "import numpy"
       pip install %ONNX_PATH%
       pip uninstall -y protobuf
-      pip install protobuf
+      pip install "protobuf<4.21.0"
       pip install h5py==2.9.0
       pip install parameterized
       pip install %TENSORFLOW_PATH%

diff --git a/ci_build/azure_pipelines/templates/setup.yml b/ci_build/azure_pipelines/templates/setup.yml
@@ -6,6 +6,10 @@ steps:
     pip install pytest pytest-cov pytest-runner coverage graphviz requests pyyaml pillow pandas parameterized
     pip install $(CI_PIP_TF_NAME) $(CI_PIP_ONNX_NAME)
 
+    # protobuf release version 4.21.0 has some Python changes which is not compatible with tensorflow so far.
+    pip uninstall -y protobuf
+    pip install "protobuf<4.21.0"
+
     # TF < 2.7 reuires numpy <= 1.19, but onnxruntime >= 1.11 requires numpy >= 1.21
     if [[ $CI_TF_VERSION < 2.7 ]] && [[ $CI_ONNX_BACKEND == "onnxruntime" ]] ;
     then