Skip to content

Commit eb512a2

Browse files
committed
Linting fix
1 parent c4cf1cc commit eb512a2

File tree

1 file changed

+3
-3
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl

1 file changed

+3
-3
lines changed

py/torch_tensorrt/dynamo/conversion/impl/select.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -87,9 +87,9 @@ def gather(
8787
indices_tensor = []
8888

8989
for i, ind in enumerate(index):
90-
indices_tensor.append(get_trt_tensor(
91-
ctx, ind, name + f"_parameter_to_fp32_tensor_{i}"
92-
))
90+
indices_tensor.append(
91+
get_trt_tensor(ctx, ind, name + f"_parameter_to_fp32_tensor_{i}")
92+
)
9393
gather_layer = ctx.net.add_gather(input, indices_tensor, dim)
9494
set_layer_name(gather_layer, target, name + "_gather", source_ir)
9595
return gather_layer.get_output(0)

0 commit comments

Comments
 (0)