use latest py_version

Author: benieric

tests/integ/sagemaker/workflow/test_retry.py

@@ -149,6 +149,7 @@ def test_model_registration_with_model_repack(
     pipeline_name,
     region_name,
     pytorch_training_latest_version,
+    pytorch_training_latest_py_version
 ):
     base_dir = os.path.join(DATA_DIR, "pytorch_mnist")
     entry_point = os.path.join(base_dir, "mnist.py")
@@ -168,7 +169,7 @@ def test_model_registration_with_model_repack(
         entry_point=entry_point,
         role=role,
         framework_version=pytorch_training_latest_version,
-        py_version="py3",
+        py_version=pytorch_training_latest_py_version,
         instance_count=instance_count,
         instance_type=instance_type,
         sagemaker_session=pipeline_session,