replace all pytorch deploy to use latest inference version

Author: Chuyang Deng

tests/integ/test_git.py

@@ -51,7 +51,7 @@
 
 @pytest.mark.local_mode
 def test_github(
-    sagemaker_local_session, pytorch_training_latest_version, pytorch_training_latest_py_version
+    sagemaker_local_session, pytorch_inference_latest_version, pytorch_inference_latest_py_version
 ):
     script_path = "mnist.py"
     git_config = {"repo": GIT_REPO, "branch": BRANCH, "commit": COMMIT}
@@ -60,8 +60,8 @@ def test_github(
         entry_point=script_path,
         role="SageMakerRole",
         source_dir="pytorch",
-        framework_version=pytorch_training_latest_version,
-        py_version=pytorch_training_latest_py_version,
+        framework_version=pytorch_inference_latest_version,
+        py_version=pytorch_inference_latest_py_version,
         instance_count=1,
         instance_type="local",
         sagemaker_session=sagemaker_local_session,

tests/integ/test_pytorch.py

@@ -95,13 +95,13 @@ def test_fit_deploy(
 
 @pytest.mark.local_mode
 def test_local_fit_deploy(
-    sagemaker_local_session, pytorch_training_latest_version, pytorch_training_latest_py_version
+    sagemaker_local_session, pytorch_inference_latest_version, pytorch_inference_latest_py_version
 ):
     pytorch = PyTorch(
         entry_point=MNIST_SCRIPT,
         role="SageMakerRole",
-        framework_version=pytorch_training_latest_version,
-        py_version=pytorch_training_latest_py_version,
+        framework_version=pytorch_inference_latest_version,
+        py_version=pytorch_inference_latest_py_version,
         instance_count=1,
         instance_type="local",
         sagemaker_session=sagemaker_local_session,

tests/integ/test_tuner.py

@@ -748,8 +748,8 @@ def test_tuning_chainer(
 def test_attach_tuning_pytorch(
     sagemaker_session,
     cpu_instance_type,
-    pytorch_training_latest_version,
-    pytorch_training_latest_py_version,
+    pytorch_inference_latest_version,
+    pytorch_inference_latest_py_version,
 ):
     mnist_dir = os.path.join(DATA_DIR, "pytorch_mnist")
     mnist_script = os.path.join(mnist_dir, "mnist.py")
@@ -758,8 +758,8 @@ def test_attach_tuning_pytorch(
         entry_point=mnist_script,
         role="SageMakerRole",
         instance_count=1,
-        framework_version=pytorch_training_latest_version,
-        py_version=pytorch_training_latest_py_version,
+        framework_version=pytorch_inference_latest_version,
+        py_version=pytorch_inference_latest_py_version,
         instance_type=cpu_instance_type,
         sagemaker_session=sagemaker_session,
     )