update py version

Author: JingyaHuang

tests/conftest.py

@@ -273,7 +273,7 @@ def huggingface_pytorch_training_version(huggingface_training_version):
 
 @pytest.fixture(scope="module")
 def huggingface_pytorch_training_py_version(huggingface_pytorch_training_version):
-    return "py38" if Version(huggingface_pytorch_training_version) >= Version("1.9") else "py36"
+    return "py39" if Version(huggingface_pytorch_training_version) >= Version("1.9") else "py36"
 
 
 @pytest.fixture(scope="module")
@@ -329,7 +329,7 @@ def huggingface_pytorch_latest_training_py_version(
     huggingface_training_pytorch_latest_version,
 ):
     return (
-        "py38" if Version(huggingface_training_pytorch_latest_version) >= Version("1.9") else "py36"
+        "py39" if Version(huggingface_training_pytorch_latest_version) >= Version("1.9") else "py36"
     )