Skip to content

Commit 03734e4

Browse files
committed
Fix gru test.
1 parent 5bd2da8 commit 03734e4

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/onnx2pytorch/convert/test_gru.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,9 @@ def test_gru_linear_before_reset(linear_before_reset, bidirectional):
139139
[W_initializer, R_initializer, B_initializer],
140140
)
141141

142-
model = helper.make_model(graph, opset_imports=[helper.make_opsetid("", 14)])
142+
model = helper.make_model(
143+
graph, opset_imports=[helper.make_opsetid("", 14)], ir_version=8
144+
)
143145
onnx.checker.check_model(model)
144146

145147
# Run with onnxruntime to get expected output

0 commit comments

Comments
 (0)