lint

Author: metascroy

exir/emit/_emitter.py

@@ -946,9 +946,7 @@ def _emit_view(self, args: Tuple[_Argument, ...]) -> _EmitterValue:
         # Elide the view if it is static and memory planned
         spec = self.node.meta["spec"]
         is_static = spec.is_static_shape_tensor
-        is_memory_planned = (spec.mem_id is not None) and (
-            spec.mem_offset is not None
-        )
+        is_memory_planned = (spec.mem_id is not None) and (spec.mem_offset is not None)
         is_memory_planned = is_memory_planned or (
             spec.const and spec.storage is not None
         )

exir/emit/test/test_emit.py

@@ -339,20 +339,17 @@ def expected_view_ops(config):
             else:
                 return ["aten::view_copy"]
 
-
         for opname in removed_ops:
             self.assertEqual(
                 self.count_node(edge.exported_program().graph_module, opname), 0
             )
         for opname in expected_ops:
-                self.assertTrue(
-                    self.count_node(edge.exported_program().graph_module, opname) >= 1
-                )
+            self.assertTrue(
+                self.count_node(edge.exported_program().graph_module, opname) >= 1
+            )
 
         for remove_view_copy in [True, False]:
-            config = exir.ExecutorchBackendConfig(
-                remove_view_copy=remove_view_copy
-            )
+            config = exir.ExecutorchBackendConfig(remove_view_copy=remove_view_copy)
             edge_copy = deepcopy(edge)
             program = edge_copy.to_executorch(config=config).executorch_program
             for opname in removed_ops:
@@ -363,7 +360,10 @@ def expected_view_ops(config):
                 self.assertTrue(
                     any(op.name == opname for op in program.execution_plan[0].operators)
                 )
-            self.assertTrue(len(program.execution_plan[0].operators) == len(expected_ops + expected_view_ops(config)))
+            self.assertTrue(
+                len(program.execution_plan[0].operators)
+                == len(expected_ops + expected_view_ops(config))
+            )
 
     def test_operators_unique(self) -> None:
         class OpRepeatedModule(torch.nn.Module):

exir/tests/test_remove_view_copy.py

@@ -205,7 +205,7 @@ def test_spec(self) -> None:
         self.assertEqual(
             instructions[4].instr_args.op_index, 2
         )  # aten:view_copy @ idx11
-    
+
     def test_elide_static_views_does_not_remove_dynamic_views(self) -> None:
         class TestModel(nn.Module):
             def __init__(self):
@@ -215,12 +215,14 @@ def forward(self, x):
                 x = x + x
                 x = x.view(-1, 1)
                 return 2 * x
-            
+
         model = TestModel()
         model.eval()
         example_inputs = (torch.rand(5, 6),)
         dynamic_shapes = {"x": {0: torch.export.Dim("dim0", min=1, max=10)}}
-        ep = torch.export.export(model, example_inputs, strict=True, dynamic_shapes=dynamic_shapes)
+        ep = torch.export.export(
+            model, example_inputs, strict=True, dynamic_shapes=dynamic_shapes
+        )
         etpm = to_edge(ep).to_executorch(
             config=ExecutorchBackendConfig(
                 remove_view_copy=True,
@@ -230,5 +232,3 @@ def forward(self, x):
         plan = etpm.executorch_program.execution_plan[0]
         op_names = [op.name for op in plan.operators]
         self.assertTrue("executorch_prim::et_view" in op_names)
-
-