pylint fix

apache · Jan 29, 2023 · 2ecdfb4 · 2ecdfb4
1 parent bf18552
commit 2ecdfb4
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py
@@ -7787,8 +7787,10 @@ def verify_sequence_ops(tensor_shape, num_tensors, axis=0, position=0, new_axis=
             output_shape[axis] = num_tensors + 1
         else:
             output_shape[axis] = (num_tensors + 1) * output_shape[axis]
-        graph_outputs = [helper.make_tensor_value_info("output", TensorProto.FLOAT, output_shape),
-                         helper.make_tensor_value_info("output_2", TensorProto.INT, ())]
+        graph_outputs = [
+            helper.make_tensor_value_info("output", TensorProto.FLOAT, output_shape),
+            helper.make_tensor_value_info("output_2", TensorProto.INT64, []),
+        ]
 
         graph_nodes = [position_node, construct_node, insert_node, concat_node, split_node, at_node]