Skip to content

Commit 0bb2231

Browse files
authored
Update to opset 13 (#437)
1 parent aa69a3c commit 0bb2231

File tree

7 files changed

+30
-2
lines changed

7 files changed

+30
-2
lines changed

.azure-pipelines/linux-CI-nightly.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,11 @@ jobs:
1818
ONNX_PATH: onnx==1.7.0
1919
ORT_PATH: -i https://test.pypi.org/simple/ ort-nightly
2020
COREML_PATH: git+https://github.com/apple/[email protected]
21+
Python37-nightly:
22+
python.version: '3.7'
23+
ONNX_PATH: onnx==1.8.0
24+
ORT_PATH: -i https://test.pypi.org/simple/ ort-nightly
25+
COREML_PATH: git+https://github.com/apple/[email protected]
2126
maxParallel: 3
2227

2328
steps:

.azure-pipelines/linux-conda-CI.yml

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,12 @@ jobs:
4343
ONNXRT_PATH: onnxruntime==1.3.0
4444
COREML_PATH: git+https://github.com/apple/[email protected]
4545
xgboost.version: '>=1.0'
46+
Python37-180-RT160:
47+
python.version: '3.7'
48+
ONNX_PATH: onnx==1.8.0
49+
ONNXRT_PATH: onnxruntime==1.6.0
50+
COREML_PATH: git+https://github.com/apple/[email protected]
51+
xgboost.version: '>=1.0'
4652
maxParallel: 3
4753

4854
steps:

.azure-pipelines/win32-CI-nightly.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,11 @@ jobs:
1818
ONNX_PATH: onnx==1.7.0
1919
ONNXRT_PATH: -i https://test.pypi.org/simple/ ort-nightly
2020
COREML_PATH: git+https://github.com/apple/[email protected]
21+
Python37-nightly:
22+
python.version: '3.7'
23+
ONNX_PATH: onnx==1.8.0
24+
ONNXRT_PATH: -i https://test.pypi.org/simple/ ort-nightly
25+
COREML_PATH: git+https://github.com/apple/[email protected]
2126
maxParallel: 3
2227

2328
steps:

.azure-pipelines/win32-conda-CI.yml

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,6 +48,13 @@ jobs:
4848
COREML_PATH: git+https://github.com/apple/[email protected]
4949
sklearn.version: ''
5050

51+
Python37-180-RT160:
52+
python.version: '3.7'
53+
ONNX_PATH: onnx==1.8.0
54+
ONNXRT_PATH: onnxruntime==1.6.0
55+
COREML_PATH: git+https://github.com/apple/[email protected]
56+
sklearn.version: ''
57+
5158
maxParallel: 3
5259

5360
steps:

onnxmltools/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@
99
This framework converts any machine learned model into onnx format
1010
which is a common language to describe any machine learned model.
1111
"""
12-
__version__ = "1.7.0"
12+
__version__ = "1.8.0"
1313
__author__ = "Microsoft"
1414
__producer__ = "OnnxMLTools"
1515
__producer_version__ = __version__

onnxmltools/convert/lightgbm/convert.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66

77
from uuid import uuid4
88
import lightgbm
9+
import warnings
910
from onnxconverter_common.onnx_ex import get_maximum_opset_supported
1011
import onnx
1112
from ..common._topology import convert_topology
@@ -61,6 +62,10 @@ def convert(model, name=None, initial_types=None, doc_string='', target_opset=No
6162
from hummingbird.ml import convert
6263
from hummingbird.ml import constants
6364

65+
if target_opset == 13:
66+
warnings.warn('Pytorch-onnx does not support opset 13 yet, use opset 12 instead.')
67+
target_opset = 12
68+
6469
extra_config = {}
6570
extra_config[constants.ONNX_INITIAL_TYPES] = initial_types
6671
extra_config[constants.ONNX_OUTPUT_MODEL_NAME] = name

requirements.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
keras2onnx
22
numpy
33
onnx
4-
onnxconverter-common>=1.7.0, <1.8.0
4+
onnxconverter-common>=1.8.0, <1.9.0
55
protobuf
66
skl2onnx

0 commit comments

Comments
 (0)