We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent bf203e3 commit 238f1b8Copy full SHA for 238f1b8
tests/conftest.py
@@ -221,12 +221,14 @@ def mxnet_eia_latest_py_version():
221
222
@pytest.fixture(scope="module", params=["py2", "py3"])
223
def pytorch_training_py_version(pytorch_training_version, request):
224
- if Version(pytorch_training_version) < Version("1.5.0"):
225
- return request.param
+ if Version(pytorch_training_version) >= Version("1.13"):
+ return "py39"
226
elif Version(pytorch_training_version) >= Version("1.9"):
227
return "py38"
228
- else:
+ elif Version(pytorch_training_version) >= Version("1.5"):
229
return "py3"
230
+ else:
231
+ return request.param
232
233
234
0 commit comments