Skip to content

Commit 1396f35

Browse files
committed
lower the min_torch version in unittest.
1 parent aca9fd1 commit 1396f35

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/tests_pytorch/models/test_onnx.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -145,8 +145,8 @@ def test_error_if_no_input(tmp_path):
145145
"dynamo",
146146
[
147147
None,
148-
pytest.param(False, marks=RunIf(min_torch="2.7.0", dynamo=True, onnxscript=True)),
149-
pytest.param(True, marks=RunIf(min_torch="2.7.0", dynamo=True, onnxscript=True)),
148+
pytest.param(False, marks=RunIf(min_torch="2.6.0", dynamo=True, onnxscript=True)),
149+
pytest.param(True, marks=RunIf(min_torch="2.6.0", dynamo=True, onnxscript=True)),
150150
],
151151
)
152152
@RunIf(onnx=True)
@@ -184,7 +184,7 @@ def to_numpy(tensor):
184184
assert np.allclose(to_numpy(torch_out), ort_outs[0], rtol=1e-03, atol=1e-05)
185185

186186

187-
@RunIf(min_torch="2.7.0", dynamo=True)
187+
@RunIf(min_torch="2.6.0", dynamo=True)
188188
@pytest.mark.skipif(_ONNXSCRIPT_AVAILABLE, reason="Run this test only if onnxscript is not available.")
189189
def test_model_onnx_export_missing_onnxscript():
190190
"""Test that an error is raised if onnxscript is not available."""
@@ -198,7 +198,7 @@ def test_model_onnx_export_missing_onnxscript():
198198
model.to_onnx(dynamo=True)
199199

200200

201-
@RunIf(onnx=True, min_torch="2.7.0", dynamo=True, onnxscript=True)
201+
@RunIf(onnx=True, min_torch="2.6.0", dynamo=True, onnxscript=True)
202202
def test_model_return_type():
203203
model = BoringModel()
204204
model.example_input_array = torch.randn((1, 32))

0 commit comments

Comments
 (0)