fix import

onnx · Apr 15, 2020 · 5222349 · 5222349
1 parent d481467
commit 5222349
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/onnxmltools/convert/h2o/convert.py b/onnxmltools/convert/h2o/convert.py
@@ -10,7 +10,7 @@
 import h2o
 
 from onnxconverter_common.onnx_ex import get_maximum_opset_supported
-from ...proto import onnx
+import onnx
 from ..common._topology import convert_topology
 from ..common.data_types import FloatTensorType
 from ._parse import parse_h2o

diff --git a/tests/coreml/test_cml_AllNeuralNetworkConverters.py b/tests/coreml/test_cml_AllNeuralNetworkConverters.py
@@ -8,12 +8,13 @@
         setattr(sklearn.preprocessing, 'Imputer', Imputer)
 except ImportError:
     from sklearn.preprocessing import Imputer
+import onnx
 from coremltools.models.neural_network import NeuralNetworkBuilder
 from coremltools.models import datatypes
 from coremltools.proto.FeatureTypes_pb2 import ImageFeatureType
 from distutils.version import StrictVersion
 from onnxmltools import convert_coreml
-from onnxmltools.proto import onnx
+
 
 class TestNeuralNetworkLayerConverter(unittest.TestCase):