add a todo to conftest to get latest py_version

Author: BruceZhang@eitug

tests/conftest.py

@@ -318,6 +318,21 @@ def huggingface_pytorch_latest_training_py_version(
     )
 
 
+@pytest.fixture(scope="module")
+def pytorch_training_compiler_py_version(
+    pytorch_training_compiler_version,
+):
+    return "py39" if Version(pytorch_training_compiler_version) > Version("1.12") else "py38"
+
+
+# TODO:
+# @pytest.fixture(scope="module")
+# def pytorch_training_compiler_latest_py_version(
+#     pytorch_training_compiler_latest_version,
+# ):
+#     return "py39" if Version(pytorch_training_compiler_latest_version) > Version("1.12") else "py38"
+
+
 @pytest.fixture(scope="module")
 def huggingface_pytorch_latest_inference_py_version(
     huggingface_inference_pytorch_latest_version,

tests/unit/sagemaker/training_compiler/test_pytorch_compiler.py

@@ -61,13 +61,6 @@ def cpu_instance_type():
     return "ml.m5.xlarge"
 
 
-@pytest.fixture(scope="module")
-def pytorch_training_compiler_py_version(
-    pytorch_training_compiler_version,
-):
-    return "py39" if Version(pytorch_training_compiler_version) > Version("1.12") else "py38"
-
-
 @pytest.fixture(name="sagemaker_session", scope="function")
 def fixture_sagemaker_session():
     boto_mock = Mock(name="boto_session", region_name=REGION)