Merge pull request #1843 from fatcat-z/fix_ci_issue

Specify the version scope of tensorflow-text and numpy.
onnx · Feb 15, 2022 · 872a96b · 872a96b
2 parents 0da3c78 + ae4ca68
commit 872a96b
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 3 deletions.
diff --git a/ci_build/azure_pipelines/templates/keras2onnx_application_tests.yml b/ci_build/azure_pipelines/templates/keras2onnx_application_tests.yml
@@ -45,6 +45,7 @@ steps:
       $(INSTALL_TRANSFORMERS)
       pip install keras-self-attention
       pip install pytest pytest-cov pytest-runner
+      pip install numpy==1.19
     displayName: 'Install dependencies'
 
   - script: |
@@ -101,6 +102,7 @@ steps:
       %INSTALL_TRANSFORMERS%
       pip install keras-self-attention
       pip install pytest pytest-cov pytest-runner
+      pip install numpy==1.19
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/ci_build/azure_pipelines/templates/keras2onnx_unit_test.yml b/ci_build/azure_pipelines/templates/keras2onnx_unit_test.yml
@@ -32,6 +32,7 @@ steps:
       pip install -r requirements-dev.txt
       pip install pytest pytest-cov pytest-runner
       $(INSTALL_ORT)
+      pip install numpy==1.19
     displayName: 'Install dependencies'
 
   - script: |
@@ -77,6 +78,7 @@ steps:
       pip install git+https://github.com/microsoft/onnxconverter-common
       pip install pytest pytest-cov pytest-runner
       %INSTALL_ORT%
+      pip install numpy==1.19
     displayName: 'Install dependencies'
 
   - script: |

diff --git a/ci_build/azure_pipelines/templates/setup.yml b/ci_build/azure_pipelines/templates/setup.yml
@@ -39,15 +39,15 @@ steps:
       fi
       if [[ $CI_TF_VERSION == 2.5* ]] ;
       then
-        pip install tensorflow-text>=2.5
+        pip install "tensorflow-text>=2.5,<2.6"
       fi
       if [[ $CI_TF_VERSION == 2.6* ]] ;
       then
-        pip install tensorflow-text>=2.6
+        pip install "tensorflow-text>=2.6,<2.7"
       fi
       if [[ $CI_TF_VERSION == 2.7* ]] ;
       then
-        pip install tensorflow-text>=2.6
+        pip install "tensorflow-text>=2.7,<2.8"
       fi
     fi