flake8 & clang

pytorch · Oct 3, 2020 · bc02b94 · bc02b94
1 parent 6268036
commit bc02b94
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/test/onnx/test_pytorch_onnx_onnxruntime.py b/test/onnx/test_pytorch_onnx_onnxruntime.py
@@ -714,7 +714,9 @@ def forward(self, x):
 
         x2 = [] if x2 is None else [x2]
         if len(x2) > 0:
-            self.run_test(Squeeze(d), x1, input_names=['input'], dynamic_axes={'input': {0: '0', 1: '1', 2: '2'}}, test_with_inputs=x2)
+            self.run_test(Squeeze(d), x1,
+                          input_names=['input'], dynamic_axes={'input': {0: '0', 1: '1', 2: '2'}},
+                          test_with_inputs=x2)
         else:
             self.run_test(Squeeze(d), x1)
 

diff --git a/torch/csrc/jit/passes/onnx/shape_type_inference.cpp b/torch/csrc/jit/passes/onnx/shape_type_inference.cpp
@@ -213,14 +213,16 @@ Node* CloneNodeToGraph(Node* n, std::shared_ptr<Graph> n_graph) {
         // order to be consumed as inputs
         std::vector<Value*> unsqueezed;
         for (auto* input : lc_node->inputs()) {
-          Node* unsqueezed_node = n_graph->insertNode(n_graph->create(::c10::onnx::Unsqueeze, 1));
+          Node* unsqueezed_node =
+              n_graph->insertNode(n_graph->create(::c10::onnx::Unsqueeze, 1));
           auto new_input = n_graph->addInput();
           new_input->copyMetadata(input);
           unsqueezed_node->addInput(new_input);
           unsqueezed_node->is_(attr::axes, {0});
           unsqueezed.emplace_back(unsqueezed_node->output());
         }
-        Node* concat_node = n_graph->insertNode(n_graph->create(::c10::onnx::Concat, 1));
+        Node* concat_node =
+            n_graph->insertNode(n_graph->create(::c10::onnx::Concat, 1));
         concat_node->i_(attr::axis, 0);
         for (auto v : unsqueezed) {
           concat_node->addInput(v);