Fix linter (pytorch#10404)

Author: kirklandsign

backends/cadence/aot/fuse_ops.py

@@ -528,7 +528,9 @@ class FuseCascadedViewOps(ExportPass):
 
     def fuse_cascaded_view_ops(self, graph_module: torch.fx.GraphModule):
         view_target = exir_ops.edge.aten.view_copy.default
-        for view_node in graph_module.graph.find_nodes(op="call_function", target=view_target, sort=True):
+        for view_node in graph_module.graph.find_nodes(
+            op="call_function", target=view_target, sort=True
+        ):
             input_view = view_node.args[0]
             if input_view.op != "call_function" or input_view.target != view_target:
                 continue

backends/cadence/aot/replace_ops.py

@@ -2259,7 +2259,6 @@ def call(self, graph_module: torch.fx.GraphModule) -> PassResult:
         return result
 
 
-
 @register_cadence_pass(CadencePassAttribute(opt_level=1))
 class ReplacePowWithMullPass(ExportPass):
     """
@@ -2274,9 +2273,13 @@ def call_operator(
         meta: NodeMetadata,
     ) -> ProxyValue:
         # TODO(eigen): Add support for other degrees.
-        if op not in {
-            exir_ops.edge.aten.pow.Scalar,
-        } or args[0] != 2:
+        if (
+            op
+            not in {
+                exir_ops.edge.aten.pow.Scalar,
+            }
+            or args[0] != 2
+        ):
             return super().call_operator(op, args, kwargs, meta)
 
         return super().call_operator(

backends/cadence/aot/tests/test_replace_ops_passes.py

@@ -30,13 +30,13 @@
     ReplaceEmptyTensorsWithFullPass,
     ReplaceFunctionallyEquivalentOpTargets,
     ReplaceGeluWithApproximateGeluPass,
-    ReplacePowWithMullPass,
     ReplaceIm2RowWithViewPass,
     ReplaceLinearWithFullyConnectedOpPass,
     ReplaceMMWithAddMMPass,
     ReplaceNopTransposeOrPermuteWithViewPass,
     ReplacePadWithCatPass,
     ReplacePermuteWithTransposePass,
+    ReplacePowWithMullPass,
     ReplaceRepeatWithCatPass,
     ReplaceScalarTensorWithFullPass,
     ReplaceScalarWithTensorArgPass,
@@ -1338,7 +1338,7 @@ def test_replace_split_with_sizes_with_slice(self):
     def test_replace_pow_with_mul(self):
         class Pow(torch.nn.Module):
             def forward(self, input):
-                return  torch.ops.aten.pow.Scalar(2, input)
+                return torch.ops.aten.pow.Scalar(2, input)
 
         input = torch.randn(2, 1, 64)
 
@@ -1347,7 +1347,6 @@ def forward(self, input):
         p = ReplacePowWithMullPass()
         graph_after_passes = cast(PassResult, p(graph_module)).graph_module
 
-
         self.assertEqual(
             count_node(
                 graph_after_passes,