forked from onnx/onnxmltools
-
Notifications
You must be signed in to change notification settings - Fork 0
145 lines (123 loc) · 5.22 KB
/
ci.yml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
name: CI
on: [push, pull_request]
jobs:
run:
name: ${{ matrix.os }} py==${{ matrix.python_version }} - sklearn${{ matrix.sklearn_version }} - ${{ matrix.onnxrt_version }}
runs-on: ${{ matrix.os }}
strategy:
matrix:
os: [ubuntu-latest, macos-latest, windows-latest]
python_version: ['3.12', '3.11', '3.10', '3.9']
include:
- python_version: '3.12'
documentation: 0
numpy_version: '>=1.21.1'
scipy_version: '>=1.7.0'
onnx_version: 'onnx==1.16.0'
onnxrt_version: 'onnxruntime==1.18.0'
sklearn_version: '==1.4.2'
lgbm_version: ">=4"
xgboost_version: ">=2"
- python_version: '3.11'
documentation: 1
numpy_version: '>=1.21.1'
scipy_version: '>=1.7.0'
onnx_version: 'onnx<1.16.0'
onnxrt_version: 'onnxruntime==1.17.3'
sklearn_version: '==1.4.2'
lgbm_version: ">=4"
xgboost_version: ">=2"
- python_version: '3.10'
documentation: 0
numpy_version: '>=1.21.1'
scipy_version: '>=1.7.0'
onnx_version: 'onnx==1.14.1'
onnxrt_version: 'onnxruntime==1.16.3'
sklearn_version: '==1.4.2'
lgbm_version: "<4"
xgboost_version: "<2"
- python_version: '3.9'
documentation: 0
numpy_version: '>=1.21.1'
scipy_version: '>=1.7.0'
onnx_version: 'onnx==1.13.0'
onnxrt_version: 'onnxruntime==1.14.1'
sklearn_version: '==1.4.2'
lgbm_version: "<4"
xgboost_version: "<2"
steps:
- name: Checkout repository
uses: actions/checkout@v4
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python_version }}
- name: Install requirements
run: python -m pip install -r requirements.txt
- name: Install requirements dev
run: python -m pip install -r requirements-dev.txt
- name: Cache pip
uses: actions/cache@v4
with:
path: ~/.cache/pip
key: ${{ runner.os }}-pip-${{ hashFiles('requirements-dev.txt') }}
restore-keys: |
${{ runner.os }}-pip-
${{ runner.os }}-
- name: Install specific versions
run: pip install "${{ matrix.onnx_version }}" "${{ matrix.onnxrt_version }}" "numpy${{ matrix.numpy_version }}" "scikit-learn${{ matrix.sklearn_version}}" "scipy${{ matrix.scipy_version }}" "xgboost${{ matrix.xgboost_version }}"
- name: Install lightgbm
if: matrix.os != 'macos-latest'
run: pip install "lightgbm${{ matrix.lgbm_version }}"
- name: Install
run: pip install -e .
- name: versions
run: |
python -c "from numpy import __version__;print('numpy', __version__)"
python -c "from pandas import __version__;print('pandas', __version__)"
python -c "from scipy import __version__;print('scipy', __version__)"
python -c "from sklearn import __version__;print('sklearn', __version__)"
python -c "from onnxruntime import __version__;print('onnxruntime', __version__)"
python -c "from onnx import __version__;print('onnx', __version__)"
python -c "from xgboost import __version__;print('xgboost', __version__)"
python -c "from catboost import __version__;print('catboost', __version__)"
python -c "import onnx.defs;print('onnx_opset_version', onnx.defs.onnx_opset_version())"
- name: versions lightgbm
if: matrix.os != 'macos-latest'
run: |
python -c "from lightgbm import __version__;print('lightgbm', __version__)"
python -c "from hummingbirdml import __version__;print('hummingbirdml', __version__)"
- name: Run tests baseline
run: pytest --maxfail=10 --durations=10 tests/baseline
- name: Run tests utils
run: pytest --maxfail=10 --durations=10 tests/utils
- name: Run tests catboost
run: pytest --maxfail=10 --durations=10 tests/catboost
- name: Run tests lightgbm
if: matrix.os != 'macos-latest'
run: pytest --maxfail=10 --durations=10 tests/lightgbm
- name: Run tests xgboost
run: pytest --maxfail=10 --durations=10 tests/xgboost
- name: Run tests svmlib
run: pytest --maxfail=10 --durations=10 tests/svmlib
- name: Run tests h2o
if: matrix.os != 'macos-latest'
run: |
pip install h2o
pytest --maxfail=10 --durations=10 tests/h2o
- name: Run tests pysparkml
if: matrix.os != 'macos-latest'
run: pytest --maxfail=10 --durations=10 tests/sparkml
- name: Run tests hummingbirdml
if: matrix.os != 'macos-latest'
run: |
pip install torch --extra-index-url https://download.pytorch.org/whl/cpu
pip install hummingbird-ml --no-deps
pytest --maxfail=10 --durations=10 tests/hummingbirdml
- name: Run tests baseline
run: pytest --maxfail=10 --durations=10 tests/baseline
- name: build
run: pip wheel .
- uses: actions/upload-artifact@v4
with:
path: ./dist/**