lint

sdpython · Aug 3, 2021 · c0e2479 · c0e2479
1 parent 50373e6
commit c0e2479
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/mlprodict/onnx_conv/onnx_ops/onnx_complex.py b/mlprodict/onnx_conv/onnx_ops/onnx_complex.py
@@ -1,8 +1,7 @@
 """
 @file
-@brief Custom operators for FFT.
+@brief Custom operators for complex numbers.
 """
-import numpy
 from skl2onnx.algebra.onnx_operator import OnnxOperator
 
 

diff --git a/mlprodict/onnxrt/ops_cpu/op_complex_abs.py b/mlprodict/onnxrt/ops_cpu/op_complex_abs.py
@@ -21,7 +21,7 @@ def _find_custom_operator_schema(self, op_name):
         raise RuntimeError(  # pragma: no cover
             "Unable to find a schema for operator '{}'.".format(op_name))
 
-    def _run(self, x):  # pylint: disable=W0221        
+    def _run(self, x):  # pylint: disable=W0221
         y = numpy.absolute(x)
         if x.dtype == numpy.complex64:
             y = y.astype(numpy.float32)