Rsqrt and linting error

Author: apbose

py/torch_tensorrt/fx/converters/aten_ops_converters.py

@@ -327,11 +327,11 @@ def aten_ops_rsqrt(
     kwargs: Dict[str, Argument],
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
-    
+
     return rsqrt(
-        network, 
-        target, 
-        SourceIR.ATEN, 
+        network,
+        target,
+        SourceIR.ATEN,
         name,
         args[0],
     )

py/torch_tensorrt/fx/converters/impl/elementwise/ops.py

@@ -117,25 +117,25 @@ def rsqrt(
     source_ir: Optional[SourceIR],
     name: str,
     input: TRTTensor,
-    other: TRTTensor,
 ) -> TRTTensor:
-    
+
     sqrt_trt_output = convert_unary(
         network,
         target,
         source_ir,
-        f"{name}"_sqrt,
+        f"{name}_sqrt",
         trt.UnaryOperation.SQRT,
         input,
     )
 
     output = convert_binary_elementwise(
         network,
+        target,
+        source_ir,
+        f"{name}_output",
+        trt.ElementWiseOperation.DIV,
         1,
         sqrt_trt_output,
-        trt.ElementWiseOperation.DIV,
-        target,
-        f"{name}_outpur",
     )
-    
-    return output
+
+    return output

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

@@ -5,7 +5,7 @@
 from torch_tensorrt.fx.tools.common_fx2trt import DispatchTestCase, InputTensorSpec
 
 
-class TestRSubConverter(DispatchTestCase):
+class TestRSqrtConverter(DispatchTestCase):
     @parameterized.expand(
         [
             ("2d_dim_alpha", (2, 1), 2),
@@ -26,4 +26,4 @@ def forward(self, input):
 
 
 if __name__ == "__main__":
-    run_tests()
+    run_tests()