8000 fix issue · sdpython/onnx-array-api@722fd2a · GitHub
[go: up one dir, main page]
Skip to content

Commit 722fd2a

Browse files
xaduprexadupre
committed
fix issue
1 parent 9351aca commit 722fd2a

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

‎_unittests/ut_translate_api/test_translate_builder.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ def setUp(self):
2020
self.maxDiff = None
2121

2222
def test_exp(self):
23-
onx = start(opset=19).vin("X").Exp().rename("Y").vout().to_onnx()
23+
onx = start(opset=19, ir_version=11).vin("X").Exp().rename("Y").vout().to_onnx()
2424
self.assertIsInstance(onx, ModelProto)
2525
self.assertIn("Exp", str(onx))
2626
ref = ReferenceEvaluator(onx)
@@ -69,7 +69,7 @@ def light_api(
6969

7070
def test_zdoc(self):
7171
onx = (
72-
start(opset=19)
72+
start(opset=19, ir_version=11)
7373
.vin("X")
7474
.reshape((-1, 1))
7575
.Transpose(perm=[1, 0])
@@ -119,7 +119,7 @@ def light_api(
119119
check_model(model)
120120

121121
def test_exp_f(self):
122-
onx = start(opset=19).vin("X").Exp().rename("Y").vout().to_onnx()
122+
onx = start(opset=19, ir_version=11).vin("X").Exp().rename("Y").vout().to_onnx()
123123
self.assertIsInstance(onx, ModelProto)
124124
self.assertIn("Exp", str(onx))
125125
ref = ReferenceEvaluator(onx)

0 commit comments

Comments
 (0)
0