Skip to content

GitLab

Commit 9d174e76 authored by Ted Themistokleous's avatar Ted Themistokleous
Browse files

Fix format in gen_onnx.py

parent e05b63f0
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
test/onnx/gen_onnx.py
View file @ 9d174e76
......@@ -2266,8 +2266,7 @@ def if_then_trailing_one_shape_test():
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 1])
then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.FLOAT,
[2])
onnx.TensorProto.FLOAT, [2])
else_out = onnx.helper.make_tensor_value_info('else_out',
onnx.TensorProto.FLOAT,
[2, 1])
......@@ -2578,8 +2577,7 @@ def if_else_trailing_one_shape_test():
onnx.TensorProto.FLOAT,
[2, 1])
else_out = onnx.helper.make_tensor_value_info('else_out',
onnx.TensorProto.FLOAT,
[2])
onnx.TensorProto.FLOAT, [2])
xt = np.ones((2, 1)).astype(np.float)
xt_tensor = helper.make_tensor(name='xt',
......@@ -2628,8 +2626,7 @@ def if_then_empty_shape_test():
y = onnx.helper.make_tensor_value_info('y', onnx.TensorProto.FLOAT, [2, 3])
then_out = onnx.helper.make_tensor_value_info('then_out',
onnx.TensorProto.FLOAT,
[])
onnx.TensorProto.FLOAT, [])
else_out = onnx.helper.make_tensor_value_info('else_out',
onnx.TensorProto.FLOAT,
[2, 3])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment