linter

Differential Revision: D67551054 Pull Request resolved: pytorch#7422
kirklandsign · Dec 21, 2024 · 6c3a792 · 6c3a792
1 parent c09d39b
commit 6c3a792
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/exir/emit/test/test_emit.py b/exir/emit/test/test_emit.py
@@ -335,7 +335,8 @@ def forward(self, x: torch.Tensor) -> torch.Tensor:
 
         program = (
             to_edge(export(f, (torch.randn(2, 3, 5),), strict=True))
-            .to_executorch().executorch_program
+            .to_executorch()
+            .executorch_program
         )
         exir.print_program.pretty_print(program)
 
@@ -522,8 +523,7 @@ def forward(self, x: torch.Tensor, n: torch.Tensor) -> torch.Tensor:
         program = (
             to_edge(export(f, (x, x), strict=True))
             # .to_edge(self.compile_config)  # TODO(larryliu): fix cat
-            .to_executorch()
-            .executorch_program
+            .to_executorch().executorch_program
         )
 
         self.assertEqual(len(program.execution_plan[0].chains[0].instructions), 1)

diff --git a/exir/tests/test_quant_fusion_pass.py b/exir/tests/test_quant_fusion_pass.py
@@ -163,7 +163,9 @@ def forward(self, x, y):
             exactly=True,
         ).check("executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor").check(
             "executorch_exir_dialects_edge__ops_quantized_decomposed_quantize_per_tensor_default"
-        ).check("executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor").check(
+        ).check(
+            "executorch_exir_dialects_edge__ops_aten_slice_copy_Tensor"
+        ).check(
             "executorch_exir_dialects_edge__ops_quantized_decomposed_add_default"
         ).check(
             "executorch_exir_dialects_edge__ops_quantized_decomposed_dequantize_per_tensor_default"
@@ -175,9 +177,7 @@ def forward(self, x, y):
         # check that we are using out variant of add and slice_copy
         FileCheck().check("torch.ops.quantized_decomposed.add.out").check(
             "torch.ops.aten.slice_copy.Tensor_out"
-        ).run(
-            m.exported_program().graph_module.code
-        )
+        ).run(m.exported_program().graph_module.code)
 
     def test_cat(self) -> None:
         class M(torch.nn.Module):