Skip to content

Commit 4a32b90

Browse files
committed
linting
1 parent deec17e commit 4a32b90

File tree

2 files changed

+6
-10
lines changed

2 files changed

+6
-10
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -258,13 +258,13 @@ def aten_ops_neg(
258258
args: Tuple[Argument, ...],
259259
kwargs: Dict[str, Argument],
260260
name: str,
261-
) -> Union[TRTTensor, Sequence[TRTTensor]]:
262-
261+
) -> Union[TRTTensor, Sequence[TRTTensor]]:
262+
263263
return impl.unary.neg(
264264
network,
265265
target,
266266
SourceIR.ATEN,
267-
name,
267+
name,
268268
args[0],
269269
)
270270

py/torch_tensorrt/dynamo/conversion/impl/unary/ops.py

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -97,6 +97,7 @@ def sign(
9797
1,
9898
)
9999

100+
100101
def neg(
101102
network: TRTNetwork,
102103
target: Target,
@@ -105,10 +106,5 @@ def neg(
105106
input_val: TRTTensor,
106107
) -> TRTTensor:
107108
return convert_unary(
108-
network,
109-
target,
110-
source_ir,
111-
name,
112-
trt.UnaryOperation.NEG,
113-
input_val
114-
)
109+
network, target, source_ir, name, trt.UnaryOperation.NEG, input_val
110+
)

0 commit comments

Comments
 (0)