Skip to content

Commit c70208b

Browse files
committed
Update signal.py
Signed-off-by: xavier dupré <[email protected]>
1 parent 033fa65 commit c70208b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tf2onnx/onnx_opset/signal.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,7 +101,10 @@ def DFT_real(x, fft_length=None):
101101
"""
102102
consumers = ctx.find_output_consumers(node.output[0])
103103
consumer_types = set(op.type for op in consumers)
104-
utils.make_sure(consumer_types == {'ComplexAbs'}, "Current implementation of RFFT only allows ComplexAbs as consumer not %r", consumer_types)
104+
utils.make_sure(
105+
consumer_types == {'ComplexAbs'},
106+
"Current implementation of RFFT only allows ComplexAbs as consumer not %r",
107+
consumer_types)
105108

106109
onnx_dtype = ctx.get_dtype(node.input[0])
107110
shape = ctx.get_shape(node.input[0])

0 commit comments

Comments
 (0)