Skip to content

Commit e77240a

Browse files
authored
Update ONNX_HUB_MAINIFEST with the latest files and add CI check for it (#547)
* enable partial manifest generation Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * cd workflow_scripts Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * python -m pip install typepy BeautifulSoup4 Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * pip install Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * reverse Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * git-lfs if not exist Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * change PATH to make it exec under root dir Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * use --drop instead of --keep Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * --drop Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com> * Only git-lfs when needed Signed-off-by: Chun-Wei Chen <jacky82226@gmail.com>
1 parent ce55fe1 commit e77240a

7 files changed

+2648
-723
lines changed

.github/PULL_REQUEST_TEMPLATE.md

+3
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,9 @@ https://github.com/microsoft/onnxruntime/blob/master/tools/python/PythonTools.md
128128
129129
<hr>
130130
131+
### Update ONNX_HUB_MANIFEST.json for ONNX Hub
132+
If this PR does update/add .onnx or .tar.gz files, please use `python workflow_scripts/generate_onnx_hub_manifest.py --target diff` to update ONNX_HUB_MANIFEST.json with according model information (especially SHA) for ONNX Hub.
133+
131134
### References
132135
Link to paper or references.
133136

.github/workflows/linux_ci.yml

+8-6
Original file line numberDiff line numberDiff line change
@@ -33,17 +33,19 @@ jobs:
3333
- name: Install dependencies
3434
run: |
3535
python -m pip install --upgrade pip
36-
# TODO: now ONNX only supports Protobuf <= 3.20.1
37-
python -m pip install protobuf==3.20.1
3836
python -m pip install onnx onnxruntime requests py-cpuinfo
3937
python -m cpuinfo
4038
39+
- name: Test updated ONNX_HUB_MANIFEST.json
40+
run: |
41+
python -m pip install typepy BeautifulSoup4 markdown pandas
42+
python workflow_scripts/generate_onnx_hub_manifest.py --target diff --drop
43+
git diff --exit-code -- ONNX_HUB_MANIFEST.json || { echo 'Please use "python workflow_scripts/generate_onnx_hub_manifest.py --target diff" to update ONNX_HUB_MANIFEST.json.' ; exit 1; }
44+
4145
- name: Test new models by onnx
4246
run: |
43-
python workflow_scripts/test_models.py --target onnx
47+
python workflow_scripts/test_models.py --target onnx --drop
4448
4549
- name: Test new models by onnxruntime
4650
run: |
47-
python workflow_scripts/test_models.py --target onnxruntime
48-
49-
51+
python workflow_scripts/test_models.py --target onnxruntime --drop

.github/workflows/windows_ci.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -40,8 +40,8 @@ jobs:
4040
4141
- name: Test new models by onnx
4242
run: |
43-
python workflow_scripts/test_models.py --target onnx
43+
python workflow_scripts/test_models.py --target onnx --drop
4444
4545
- name: Test new models by onnxruntime
4646
run: |
47-
python workflow_scripts/test_models.py --target onnxruntime
47+
python workflow_scripts/test_models.py --target onnxruntime --drop

0 commit comments

Comments
 (0)