fix format

microsoft · Nov 2, 2023 · 47c45aa · 47c45aa
1 parent fe11182
commit 47c45aa
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/onnxruntime/test/testdata/trt_reshape_test.py b/onnxruntime/test/testdata/trt_reshape_test.py
@@ -20,8 +20,14 @@ def generate_model(model_name):
         nodes,
         "trt_engine_wrapper",
         [  # input
-            helper.make_tensor_value_info("data", TensorProto.FLOAT, ["N",2]),
-            helper.make_tensor_value_info("shape", TensorProto.INT64, [2,]),
+            helper.make_tensor_value_info("data", TensorProto.FLOAT, ["N", 2]),
+            helper.make_tensor_value_info(
+                "shape",
+                TensorProto.INT64,
+                [
+                    2,
+                ],
+            ),
         ],
         [  # output
             helper.make_tensor_value_info("reshaped", TensorProto.FLOAT, [4, 1]),