Skip to content

Commit 4370181

Browse files
committed
add py311 to training version
1 parent ce777e6 commit 4370181

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

tests/conftest.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -293,6 +293,8 @@ def huggingface_pytorch_training_version(huggingface_training_version):
293293

294294
@pytest.fixture(scope="module")
295295
def huggingface_pytorch_training_py_version(huggingface_pytorch_training_version):
296+
if Version(huggingface_pytorch_training_version) >= Version("2.3"):
297+
return "py311"
296298
if Version(huggingface_pytorch_training_version) >= Version("2.0"):
297299
return "py310"
298300
elif Version(huggingface_pytorch_training_version) >= Version("1.13"):

0 commit comments

Comments
 (0)