Fix python lint issue

Author: nvpohanh

py/torch_tensorrt/fx/converters/acc_ops_converters.py

@@ -413,7 +413,12 @@ def acc_ops_pad_with_slice_layer(
     )
 
     shape = tuple(
-        input_shape[i] + (pad[-(i - prefix_len) * 2 - 1] + pad[-(i - prefix_len) * 2 - 2] if i >= prefix_len else 0)
+        input_shape[i]
+        + (
+            pad[-(i - prefix_len) * 2 - 1] + pad[-(i - prefix_len) * 2 - 2]
+            if i >= prefix_len
+            else 0
+        )
         for i in range(0, len(input_shape))
     )
     stride = tuple([1] * len(shape))

py/torch_tensorrt/fx/test/passes/test_fuse_permute_matmul_trt.py

@@ -37,8 +37,20 @@ class TestFusePermuteMatmul(AccTestCase):
                 lambda x: x.permute(0, 1, 3, 2),
                 torch.matmul,
             ),
-            param("transpose_lhs_bmm_broadcast", (3, 2), (3, 3, 4), tranpose_last_two_dims, op=torch.matmul),
-            param("transpose_rhs_bmm_broadcast", (3, 3, 4), (3, 4), rhs_op=tranpose_last_two_dims, op=torch.matmul),
+            param(
+                "transpose_lhs_bmm_broadcast",
+                (3, 2),
+                (3, 3, 4),
+                tranpose_last_two_dims,
+                op=torch.matmul,
+            ),
+            param(
+                "transpose_rhs_bmm_broadcast",
+                (3, 3, 4),
+                (3, 4),
+                rhs_op=tranpose_last_two_dims,
+                op=torch.matmul,
+            ),
         ]
     )
     def test_fuse_permute_matmul(