Skip to content

Commit 238f1b8

Browse files
author
Shibo Xing
committed
test: update pytorch training py_version configuration
1 parent bf203e3 commit 238f1b8

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

tests/conftest.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -221,12 +221,14 @@ def mxnet_eia_latest_py_version():
221221

222222
@pytest.fixture(scope="module", params=["py2", "py3"])
223223
def pytorch_training_py_version(pytorch_training_version, request):
224-
if Version(pytorch_training_version) < Version("1.5.0"):
225-
return request.param
224+
if Version(pytorch_training_version) >= Version("1.13"):
225+
return "py39"
226226
elif Version(pytorch_training_version) >= Version("1.9"):
227227
return "py38"
228-
else:
228+
elif Version(pytorch_training_version) >= Version("1.5"):
229229
return "py3"
230+
else:
231+
return request.param
230232

231233

232234
@pytest.fixture(scope="module", params=["py2", "py3"])

0 commit comments

Comments
 (0)