Skip to content

Commit 4dd9251

Browse files
fatcat-zfatcat-z
committed
Add keras unit tests on windows.
Signed-off-by: Jay Zhang <jiz@microsoft.com>
1 parent e8e7c00 commit 4dd9251

File tree

2 files changed

+149
-36
lines changed

2 files changed

+149
-36
lines changed

‎.github/workflows/keras_unit_test_core.yml

Lines changed: 149 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,3 +58,152 @@ jobs:
5858
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
5959
exit $status
6060
61+
run_tests_windows:
62+
strategy:
63+
matrix:
64+
tf_version: ['2.9.0', '2.13.0']
65+
python_version: ['3.8', '3.9', '3.10']
66+
opset_version: ['18', '15']
67+
ort_version: ['1.16.3']
68+
onnx_version: ['1.15.0']
69+
70+
runs-on: windows-2022
71+
72+
steps:
73+
- name: Set up Python (${{ matrix.python_version }})
74+
uses: actions/setup-python@v5
75+
with:
76+
python-version: ${{ matrix.python_version }}
77+
78+
- name: Checkout code
79+
uses: actions/checkout@v4
80+
81+
- name: Install dependencies (tf v${{ matrix.tf_version }})
82+
shell: bash
83+
run: |
84+
python -m pip install --upgrade pip
85+
pip install onnxconverter-common
86+
pip install onnx==${{ matrix.onnx_version }}
87+
pip install h5py==3.7.0
88+
pip install parameterized
89+
pip install timeout-decorator
90+
pip install coloredlogs flatbuffers
91+
pip install tensorflow==${{ matrix.tf_version }}
92+
pip install pytest pytest-cov pytest-runner
93+
pip install onnxruntime==${{ matrix.ort_version }}
94+
pip uninstall -y protobuf
95+
pip install "protobuf~=3.20"
96+
pip install -e .
97+
98+
echo "----- List all of depdencies:"
99+
pip freeze --all
100+
101+
- name: Run keras unit tests
102+
run: |
103+
set -x
104+
status=0
105+
python -c "import onnxruntime"
106+
python -c "import onnxconverter_common"
107+
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
108+
exit $status
109+
110+
111+
run_tests_linux_TF_15:
112+
strategy:
113+
matrix:
114+
tf_version: ['1.15.0']
115+
python_version: ['3.7'] # Max version that supports tf 1.15
116+
opset_version: ['18', '15']
117+
ort_version: ['1.14.1'] # Max version that supports python 3.7
118+
onnx_version: ['1.14.1'] # Max version that supports python 3.7
119+
120+
runs-on: ubuntu-latest
121+
122+
steps:
123+
- name: Set up Python (${{ matrix.python_version }})
124+
uses: actions/setup-python@v5
125+
with:
126+
python-version: ${{ matrix.python_version }}
127+
128+
- name: Checkout code
129+
uses: actions/checkout@v4
130+
131+
- name: Install dependencies (tf v${{ matrix.tf_version }})
132+
shell: bash
133+
run: |
134+
python -m pip install --upgrade pip
135+
pip install onnxconverter-common
136+
pip install onnx==${{ matrix.onnx_version }}
137+
pip uninstall -y protobuf
138+
pip install "protobuf~=3.20"
139+
pip install h5py==3.7.0
140+
pip install parameterized
141+
pip install timeout-decorator
142+
pip install coloredlogs flatbuffers
143+
pip install tensorflow==${{ matrix.tf_version }}
144+
pip install pytest pytest-cov pytest-runner
145+
pip install onnxruntime==${{ matrix.ort_version }}
146+
pip install numpy==1.19.0
147+
pip install -e .
148+
149+
echo "----- List all of depdencies:"
150+
pip freeze --all
151+
152+
- name: Run keras unit tests
153+
run: |
154+
set -x
155+
status=0
156+
python -c "import onnxruntime"
157+
python -c "import onnxconverter_common"
158+
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
159+
exit $status
160+
161+
run_tests_windows_TF_15:
162+
strategy:
163+
matrix:
164+
tf_version: ['1.15.0']
165+
python_version: ['3.7'] # Max version that supports tf 1.15
166+
opset_version: ['18', '15']
167+
ort_version: ['1.14.1'] # Max version that supports python 3.7
168+
onnx_version: ['1.14.1'] # Max version that supports python 3.7
169+
170+
runs-on: windows-2022
171+
172+
steps:
173+
- name: Set up Python (${{ matrix.python_version }})
174+
uses: actions/setup-python@v5
175+
with:
176+
python-version: ${{ matrix.python_version }}
177+
178+
- name: Checkout code
179+
uses: actions/checkout@v4
180+
181+
- name: Install dependencies (tf v${{ matrix.tf_version }})
182+
shell: bash
183+
run: |
184+
python -m pip install --upgrade pip
185+
pip install onnxconverter-common
186+
pip install onnx==${{ matrix.onnx_version }}
187+
pip uninstall -y protobuf
188+
pip install "protobuf~=3.20"
189+
pip install h5py==3.7.0
190+
pip install parameterized
191+
pip install timeout-decorator
192+
pip install coloredlogs flatbuffers
193+
pip install tensorflow==${{ matrix.tf_version }}
194+
pip install pytest pytest-cov pytest-runner
195+
pip install onnxruntime==${{ matrix.ort_version }}
196+
pip install numpy==1.19.0
197+
pip install -e .
198+
199+
echo "----- List all of depdencies:"
200+
pip freeze --all
201+
202+
- name: Run keras unit tests
203+
run: |
204+
set -x
205+
status=0
206+
python -c "import onnxruntime"
207+
python -c "import onnxconverter_common"
208+
pytest tests/keras2onnx_unit_tests --doctest-modules --junitxml=junit/test-results.xml
209+
exit $status

‎.github/workflows/main.yml

Lines changed: 0 additions & 36 deletions
This file was deleted.

0 commit comments

Comments
 (0)