Update signal.py

Signed-off-by: xavier dupré <xavier.dupre@gmail.com>
onnx · Sep 29, 2020 · c70208b · c70208b
1 parent 033fa65
commit c70208b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tf2onnx/onnx_opset/signal.py b/tf2onnx/onnx_opset/signal.py
@@ -101,7 +101,10 @@ def DFT_real(x, fft_length=None):
         """
         consumers = ctx.find_output_consumers(node.output[0])
         consumer_types = set(op.type for op in consumers)
-        utils.make_sure(consumer_types == {'ComplexAbs'}, "Current implementation of RFFT only allows ComplexAbs as consumer not %r", consumer_types)
+        utils.make_sure(
+            consumer_types == {'ComplexAbs'},
+            "Current implementation of RFFT only allows ComplexAbs as consumer not %r",
+            consumer_types)
 
         onnx_dtype = ctx.get_dtype(node.input[0])
         shape = ctx.get_shape(node.input[0])