From 7c5d4d3741f1ca9af8b770161498670d58adcf55 Mon Sep 17 00:00:00 2001 From: Jared Roesch Date: Thu, 23 May 2019 16:23:11 -0700 Subject: [PATCH] Wrap line --- python/tvm/relay/frontend/onnx.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/python/tvm/relay/frontend/onnx.py b/python/tvm/relay/frontend/onnx.py index 9336cf0ba6889..44ddb0caeeb66 100644 --- a/python/tvm/relay/frontend/onnx.py +++ b/python/tvm/relay/frontend/onnx.py @@ -1018,7 +1018,10 @@ def from_onnx(self, graph, opset): if len(array.shape) == 0 and array.dtype == 'int64': array = _nd.array(array.asnumpy().astype('int32')) self._params[node.output[0]] = array - self._nodes[node.output[0]] = new_var(node.output[0], shape=list(t_proto.dims), dtype=array.dtype) + self._nodes[node.output[0]] = new_var( + node.output[0], + shape=list(t_proto.dims), + dtype=array.dtype) else: if op_name == "ConstantFill": fill_value = attr.get('value', 0.0)