Release 0.1.0 (#236)

* Fix converter script for installation from pypi * Bump version number to 0.1.0
onnx · May 1, 2018 · 56ac628 · 56ac628
1 parent 26bc5c4
commit 56ac628
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/onnx_coreml/bin/convert.py b/onnx_coreml/bin/convert.py
@@ -4,7 +4,7 @@
 from __future__ import unicode_literals
 
 import click
-from onnx import onnx_pb2
+from onnx import onnx_pb
 from onnx_coreml import convert
 from typing import Text, IO
 
@@ -20,7 +20,7 @@
               type=str,
               help='Output path for the CoreML *.mlmodel file')
 def onnx_to_coreml(onnx_model, output):  # type: (IO[str], str) -> None
-    onnx_model_proto = onnx_pb2.ModelProto()
+    onnx_model_proto = onnx_pb.ModelProto()
     onnx_model_proto.ParseFromString(onnx_model.read())
     coreml_model = convert(onnx_model_proto)
     coreml_model.save(output)

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 import sys
 
 
-VERSION = '0.0.2'
+VERSION = '0.1.0'
 
 here = path.abspath(path.dirname(__file__))