Skip to content

Commit ba1a681

Browse files
authored
Merge branch 'master' into feature/modelcard-property
2 parents 895f099 + 2b0be8d commit ba1a681

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/conftest.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,9 @@ def pytorch_training_py_version(pytorch_training_version, request):
269269

270270
@pytest.fixture(scope="module", params=["py2", "py3"])
271271
def pytorch_inference_py_version(pytorch_inference_version, request):
272-
if Version(pytorch_inference_version) >= Version("2.0"):
272+
if Version(pytorch_inference_version) >= Version("2.3"):
273+
return "py311"
274+
elif Version(pytorch_inference_version) >= Version("2.0"):
273275
return "py310"
274276
elif Version(pytorch_inference_version) >= Version("1.13"):
275277
return "py39"

0 commit comments

Comments
 (0)