format.py

ROCm · Oct 3, 2023 · d543a26 · d543a26
1 parent 2e6923b
commit d543a26
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/test/onnx/gen_onnx.py b/test/onnx/gen_onnx.py
@@ -4947,9 +4947,8 @@ def pad_4arg_axes_test():
     x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 4, 5])
     y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 6, 4, 12])
 
-    node = onnx.helper.make_node('Pad',
-                                 inputs=['0', 'arg_pad', 'arg_val', 'arg_axes'],
-                                 outputs=['1'])
+    node = onnx.helper.make_node(
+        'Pad', inputs=['0', 'arg_pad', 'arg_val', 'arg_axes'], outputs=['1'])
 
     return ([arg_axes, arg_val, arg_pad, node], [x], [y])
 
@@ -4989,9 +4988,8 @@ def pad_4arg_neg_axes_test():
     x = helper.make_tensor_value_info('0', TensorProto.FLOAT, [1, 3, 4, 5])
     y = helper.make_tensor_value_info('1', TensorProto.FLOAT, [1, 6, 4, 12])
 
-    node = onnx.helper.make_node('Pad',
-                                 inputs=['0', 'arg_pad', 'arg_val', 'arg_axes'],
-                                 outputs=['1'])
+    node = onnx.helper.make_node(
+        'Pad', inputs=['0', 'arg_pad', 'arg_val', 'arg_axes'], outputs=['1'])
 
     return ([arg_axes, arg_val, arg_pad, node], [x], [y])