update py version for PT 2.0 tests

Author: Mike Schneider

tests/conftest.py

@@ -243,7 +243,9 @@ def mxnet_eia_latest_py_version():
 
 @pytest.fixture(scope="module", params=["py2", "py3"])
 def pytorch_training_py_version(pytorch_training_version, request):
-    if Version(pytorch_training_version) >= Version("1.13"):
+    if Version(pytorch_training_version) >= Version("2.0"):
+        return "py310"
+    elif Version(pytorch_training_version) >= Version("1.13"):
         return "py39"
     elif Version(pytorch_training_version) >= Version("1.9"):
         return "py38"
@@ -255,7 +257,9 @@ def pytorch_training_py_version(pytorch_training_version, request):
 
 @pytest.fixture(scope="module", params=["py2", "py3"])
 def pytorch_inference_py_version(pytorch_inference_version, request):
-    if Version(pytorch_inference_version) >= Version("1.13"):
+    if Version(pytorch_inference_version) >= Version("2.0"):
+        return "py310"
+    elif Version(pytorch_inference_version) >= Version("1.13"):
         return "py39"
     elif Version(pytorch_inference_version) >= Version("1.9"):
         return "py38"