Skip to content

Commit 84ff793

Browse files
apbosegs-olive
authored andcommitted
Python linting correction
1 parent a0d8245 commit 84ff793

File tree

6 files changed

+6
-8
lines changed

6 files changed

+6
-8
lines changed

py/torch_tensorrt/fx/converters/acc_ops_converters.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -1045,7 +1045,7 @@ def acc_ops_elu(
10451045
kwargs: Dict[str, Argument],
10461046
name: str,
10471047
) -> Union[TRTTensor, Sequence[TRTTensor]]:
1048-
1048+
10491049
return activation.elu(
10501050
network,
10511051
target,
@@ -1072,7 +1072,6 @@ def acc_ops_selu(
10721072
name,
10731073
kwargs["input"],
10741074
)
1075-
10761075

10771076

10781077
@tensorrt_converter(acc_ops.softsign)

py/torch_tensorrt/fx/converters/aten_ops_converters.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,7 @@ def aten_ops_elu(
179179
name: str,
180180
) -> Union[TRTTensor, Sequence[TRTTensor]]:
181181

182-
if (len(args) > 2) :
182+
if len(args) > 2:
183183
return activation.selu(
184184
network,
185185
target,

py/torch_tensorrt/fx/converters/impl/activation.py

+1-2
Original file line numberDiff line numberDiff line change
@@ -123,7 +123,6 @@ def selu(
123123
source_ir: Optional[SourceIR],
124124
name: str,
125125
input_val: TRTTensor,
126-
alpha: Optional[Any],
127126
):
128127
operation_type = trt.ActivationType.SELU
129128

@@ -138,4 +137,4 @@ def elu_dyn_range_fn(dyn_range):
138137
operation_type,
139138
input_val,
140139
dyn_range_fn=elu_dyn_range_fn,
141-
)
140+
)

py/torch_tensorrt/fx/converters/nn_ops_converters.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -51,5 +51,5 @@ def selu(network, submod, args, kwargs, layer_name):
5151
source_ir=SourceIR.NN,
5252
name=layer_name,
5353
input_val=kwargs["input"],
54-
alpha = kwargs["alpha"]
54+
alpha=kwargs["alpha"],
5555
)

py/torch_tensorrt/fx/test/converters/aten_op/test_elu_aten.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,4 +48,4 @@ def forward(self, x):
4848

4949

5050
if __name__ == "__main__":
51-
run_tests()
51+
run_tests()

py/torch_tensorrt/fx/test/converters/aten_op/test_selu_aten.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -48,4 +48,4 @@ def forward(self, x):
4848

4949

5050
if __name__ == "__main__":
51-
run_tests()
51+
run_tests()

0 commit comments

Comments
 (0)