Skip to content

Commit

Permalink
Upgrade version number, update documentation (#672)
Browse files Browse the repository at this point in the history
  • Loading branch information
xadupre authored Jun 30, 2021
1 parent 7439aa9 commit e8e12f1
Show file tree
Hide file tree
Showing 5 changed files with 7 additions and 5 deletions.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ Or you can install from the source with the latest changes.
pip install git+https://github.com/onnx/sklearn-onnx.git
```

If you install *sklearn-onnx* from its source code, you must set the environment variable `ONNX_ML=1` before installing the onnx package.
If you install *onnx* from its source code, you must set the environment variable `ONNX_ML=1` before installing the onnx package.

## Contribute
We welcome contributions in the form of feedback, ideas, or code.
Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
Expand Up @@ -4,4 +4,4 @@ scipy>=1.0
protobuf
onnx>=1.2.1
scikit-learn>=0.19
onnxconverter-common>=1.6.1,<1.9
onnxconverter-common>=1.6.1
2 changes: 1 addition & 1 deletion setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -49,8 +49,8 @@
'Operating System :: MacOS :: MacOS X',
'Operating System :: Microsoft :: Windows',
'Programming Language :: Python',
'Programming Language :: Python :: 3.6',
'Programming Language :: Python :: 3.7',
'Programming Language :: Python :: 3.8',
'Programming Language :: Python :: 3.9',
'License :: OSI Approved :: Apache Software License'],
)
3 changes: 1 addition & 2 deletions skl2onnx/__init__.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,9 @@
# SPDX-License-Identifier: Apache-2.0


"""
Main entry point to the converter from the *scikit-learn* to *onnx*.
"""
__version__ = "1.9.dev"
__version__ = "1.9"
__author__ = "Microsoft"
__producer__ = "skl2onnx"
__producer_version__ = __version__
Expand Down
3 changes: 3 additions & 0 deletions skl2onnx/common/_container.py
Original file line number Diff line number Diff line change
Expand Up @@ -386,6 +386,9 @@ def add_initializer(self, name, onnx_type, shape, content):
else:
if any(d is None for d in shape):
raise ValueError('Shape of initializer cannot contain None.')
if (hasattr(content, 'dtype') and
content.dtype in (bool, np.bool_)):
content = content.astype(np.int32)
try:
tensor = make_tensor(name, onnx_type, shape, content)
except TypeError as e:
Expand Down

0 comments on commit e8e12f1

Please sign in to comment.