Skip to content

Commit f6f8e17

Browse files
committed
fix pylint
1 parent 4cf284e commit f6f8e17

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tf2onnx/rewriter/thresholdedrelu_rewriter.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,11 +44,11 @@ def rewrite_thresholded_relu(g, ops):
4444
# if theta < 0:
4545
# continue
4646
thresholded_relu = g.make_node("ThresholdedRelu", inputs=[mul_input_edge_name], attr={"alpha": theta},
47-
shapes=[g.get_shape(mul_node.output[0])], dtypes=[g.get_dtype(mul_node.output[0])])
47+
shapes=[g.get_shape(mul_node.output[0])],
48+
dtypes=[g.get_dtype(mul_node.output[0])])
4849
ops.remove(greater_node)
4950
ops.remove(cast_node)
5051
ops.remove(mul_node)
5152
ops.append(thresholded_relu)
5253
g.replace_all_inputs(ops, mul_node.output[0], thresholded_relu.output[0])
5354
return ops
54-

0 commit comments

Comments
 (0)