Skip to content

Commit cd9f141

Browse files
authored
Specify numpy version for tf2.x version. (#1898)
Signed-off-by: Jay Zhang <[email protected]>
1 parent 8e43a17 commit cd9f141

File tree

2 files changed

+15
-5
lines changed

2 files changed

+15
-5
lines changed

ci_build/azure_pipelines/keras2onnx_application_tests.yml

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ jobs:
2020
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
2121

2222
Python37-onnx1.6:
23-
python.version: '3.7.3'
23+
python.version: '3.7'
2424
ONNX_PATH: onnx==1.6.0
2525
INSTALL_KERAS: pip install keras==2.3.1
2626
UNINSTALL_KERAS:
@@ -32,7 +32,7 @@ jobs:
3232
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
3333

3434
Python37-onnx1.9:
35-
python.version: '3.7.3'
35+
python.version: '3.7'
3636
ONNX_PATH: onnx==1.9.0
3737
INSTALL_KERAS: pip install keras==2.3.1
3838
UNINSTALL_KERAS:
@@ -44,7 +44,7 @@ jobs:
4444
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py"
4545

4646
Python37-tf2.x:
47-
python.version: '3.7.3'
47+
python.version: '3.7'
4848
ONNX_PATH: onnx==1.11.0
4949
INSTALL_KERAS:
5050
UNINSTALL_KERAS: pip uninstall keras -y
@@ -90,7 +90,7 @@ jobs:
9090
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
9191

9292
Python37-onnx1.9:
93-
python.version: '3.7.3'
93+
python.version: '3.7'
9494
ONNX_PATH: onnx==1.9.0
9595
INSTALL_KERAS: pip install keras==2.3.1
9696
UNINSTALL_KERAS:
@@ -111,7 +111,7 @@ jobs:
111111
INSTALL_KERAS_RESNET: pip install keras-resnet
112112
INSTALL_TRANSFORMERS: pip install transformers==3.4.0
113113
INSTALL_NUMPY: pip install numpy==1.19.0
114-
NIGHTLY_BUILD_TEST: python run_all.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
114+
NIGHTLY_BUILD_TEST: python run_all_v2.py --exclude "test_keras_applications_v2.py test_mask_rcnn.py"
115115

116116
steps:
117117
- template: 'templates/keras2onnx_application_tests.yml'

ci_build/azure_pipelines/keras2onnx_unit_test.yml

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,24 +21,28 @@ jobs:
2121
ONNX_PATH: onnx==1.11.0
2222
TENSORFLOW_PATH: tensorflow-cpu==2.1.0
2323
INSTALL_ORT: pip install onnxruntime==1.11.0
24+
INSTALL_NUMPY: pip install numpy==1.19.0
2425

2526
Python38-tf2.2:
2627
python.version: '3.8'
2728
ONNX_PATH: onnx==1.11.0
2829
TENSORFLOW_PATH: tensorflow-cpu==2.2.0
2930
INSTALL_ORT: pip install onnxruntime==1.11.0
31+
INSTALL_NUMPY: pip install numpy==1.19.0
3032

3133
Python38-tf2.3:
3234
python.version: '3.8'
3335
ONNX_PATH: onnx==1.11.0
3436
TENSORFLOW_PATH: tensorflow-cpu==2.3.0
3537
INSTALL_ORT: pip install onnxruntime==1.11.0
38+
INSTALL_NUMPY: pip install numpy==1.19.0
3639

3740
Python38-tf2.5:
3841
python.version: '3.8'
3942
ONNX_PATH: onnx==1.11.0
4043
TENSORFLOW_PATH: tensorflow-cpu==2.5.0
4144
INSTALL_ORT: pip install onnxruntime==1.11.0
45+
INSTALL_NUMPY: pip install numpy==1.19.0
4246

4347
############ Pure Keras Unit Tests ############
4448
Keras-Py36-tf1.15.0:
@@ -71,6 +75,7 @@ jobs:
7175
KERAS: keras==2.4.3
7276
TENSORFLOW_PATH: tensorflow==2.2.0
7377
INSTALL_ORT: pip install onnxruntime==1.11.0
78+
INSTALL_NUMPY: pip install numpy==1.19.0
7479

7580
steps:
7681
- template: 'templates/keras2onnx_unit_test.yml'
@@ -94,24 +99,28 @@ jobs:
9499
ONNX_PATH: onnx==1.11.0
95100
TENSORFLOW_PATH: tensorflow-cpu==2.1.0
96101
INSTALL_ORT: pip install onnxruntime==1.11.0
102+
INSTALL_NUMPY: pip install numpy==1.19.0
97103

98104
Python37-tf2.2:
99105
python.version: '3.7'
100106
ONNX_PATH: onnx==1.11.0
101107
TENSORFLOW_PATH: tensorflow-cpu==2.2.0
102108
INSTALL_ORT: pip install onnxruntime==1.11.0
109+
INSTALL_NUMPY: pip install numpy==1.19.0
103110

104111
Python37-tf2.3:
105112
python.version: '3.7'
106113
ONNX_PATH: onnx==1.11.0
107114
TENSORFLOW_PATH: tensorflow-cpu==2.3.0
108115
INSTALL_ORT: pip install onnxruntime==1.11.0
116+
INSTALL_NUMPY: pip install numpy==1.19.0
109117

110118
Python37-tf2.5:
111119
python.version: '3.7'
112120
ONNX_PATH: onnx==1.11.0
113121
TENSORFLOW_PATH: tensorflow-cpu==2.5.0
114122
INSTALL_ORT: pip install onnxruntime==1.11.0
123+
INSTALL_NUMPY: pip install numpy==1.19.0
115124

116125
############ Pure Keras Unit Tests ############
117126
Keras-Py36-tf1.15.0:
@@ -135,6 +144,7 @@ jobs:
135144
KERAS: keras==2.4.3
136145
TENSORFLOW_PATH: tensorflow==2.2.0
137146
INSTALL_ORT: pip install onnxruntime==1.11.0
147+
INSTALL_NUMPY: pip install numpy==1.19.0
138148

139149
steps:
140150
- template: 'templates/keras2onnx_unit_test.yml'

0 commit comments

Comments
 (0)