Run lint

Author: gramalingam

onnxscript/_internal/builder.py

@@ -214,9 +214,7 @@ def build_graph(
 class GraphBuilder:
     """Imperative builder for constructing ONNX IR graphs with automatic constant promotion, type casting, and shape inference."""
 
-    def __init__(
-        self, graph: ir.Graph, parent: GraphBuilder | None = None
-    ) -> None:
+    def __init__(self, graph: ir.Graph, parent: GraphBuilder | None = None) -> None:
         self._graph = graph
         self._parent = parent
         self._root: GraphBuilder = parent._root if parent is not None else self

onnxscript/_internal/builder_test.py

@@ -1064,7 +1064,11 @@ def test_build_graph_custom_name(self):
     def test_build_graph_with_parent(self):
         """build_graph with parent sets root on the sub-builder."""
         parent_graph = ir.Graph(
-            name="main", inputs=[], outputs=[], nodes=[], opset_imports={"": 23},
+            name="main",
+            inputs=[],
+            outputs=[],
+            nodes=[],
+            opset_imports={"": 23},
         )
         parent_builder = builder.GraphBuilder(parent_graph)
 
@@ -1083,7 +1087,11 @@ def body(op, x):
     def test_subgraph_sets_parent_and_root(self):
         """GraphBuilder.subgraph() sets parent=self on the sub-builder."""
         parent_graph = ir.Graph(
-            name="main", inputs=[], outputs=[], nodes=[], opset_imports={"": 23},
+            name="main",
+            inputs=[],
+            outputs=[],
+            nodes=[],
+            opset_imports={"": 23},
         )
         parent_builder = builder.GraphBuilder(parent_graph)
 
@@ -1097,7 +1105,11 @@ def body(op, x):
     def test_root_graph_builder_is_its_own_root(self):
         """A top-level GraphBuilder has root == self."""
         graph = ir.Graph(
-            name="main", inputs=[], outputs=[], nodes=[], opset_imports={"": 23},
+            name="main",
+            inputs=[],
+            outputs=[],
+            nodes=[],
+            opset_imports={"": 23},
         )
         gb = builder.GraphBuilder(graph)
         self.assertIs(gb.root, gb)

onnxscript/nn/_module_test.py

@@ -74,11 +74,15 @@ def test_realize_qualifies_name(self):
 
     def test_realize_in_subgraph_registers_in_root(self):
         """Parameter realized inside a subgraph builder is stored in the root graph."""
-        from onnxscript._internal.builder import GraphBuilder, build_graph
+        from onnxscript._internal.builder import GraphBuilder
         from onnxscript.onnx_types import FLOAT
 
         root_graph = ir.Graph(
-            name="main", inputs=[], outputs=[], nodes=[], opset_imports={"": 23},
+            name="main",
+            inputs=[],
+            outputs=[],
+            nodes=[],
+            opset_imports={"": 23},
         )
         root_builder = GraphBuilder(root_graph)
 
@@ -106,7 +110,11 @@ def test_realize_in_nested_subgraph_registers_in_root(self):
         from onnxscript.onnx_types import FLOAT
 
         root_graph = ir.Graph(
-            name="main", inputs=[], outputs=[], nodes=[], opset_imports={"": 23},
+            name="main",
+            inputs=[],
+            outputs=[],
+            nodes=[],
+            opset_imports={"": 23},
         )
         root_builder = GraphBuilder(root_graph)