Skip to content

Commit 426fd04

Browse files
committed
Fix python lint issue
1 parent 74f9f95 commit 426fd04

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

py/torch_tensorrt/fx/converters/acc_ops_converters.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -413,7 +413,12 @@ def acc_ops_pad_with_slice_layer(
413413
)
414414

415415
shape = tuple(
416-
input_shape[i] + (pad[-(i - prefix_len) * 2 - 1] + pad[-(i - prefix_len) * 2 - 2] if i >= prefix_len else 0)
416+
input_shape[i]
417+
+ (
418+
pad[-(i - prefix_len) * 2 - 1] + pad[-(i - prefix_len) * 2 - 2]
419+
if i >= prefix_len
420+
else 0
421+
)
417422
for i in range(0, len(input_shape))
418423
)
419424
stride = tuple([1] * len(shape))

0 commit comments

Comments
 (0)