fix

Author: pagezyhf

tests/conftest.py

@@ -363,6 +363,8 @@ def huggingface_training_compiler_pytorch_py_version(
 def huggingface_pytorch_latest_training_py_version(
     huggingface_training_pytorch_latest_version,
 ):
+    if Version(huggingface_training_pytorch_latest_version) >= Version("2.6"):
+        return "py312"
     if Version(huggingface_training_pytorch_latest_version) >= Version("2.3"):
         return "py311"
     if Version(huggingface_training_pytorch_latest_version) >= Version("2.0"):