feature: support TFS 2.2

Author: Chuyang Deng

src/sagemaker/tensorflow/defaults.py

@@ -21,7 +21,7 @@
 LATEST_VERSION = "2.2.0"
 """The latest version of TensorFlow included in the SageMaker pre-built Docker images."""
 
-LATEST_SERVING_VERSION = "2.1.0"
+LATEST_SERVING_VERSION = "2.2.0"
 """The latest version of TensorFlow Serving included in the SageMaker pre-built Docker images."""
 
 LATEST_PY2_VERSION = "2.1.0"

tests/integ/test_tf_script_mode.py

@@ -60,7 +60,7 @@ def test_mnist_with_checkpoint_config(
         sagemaker_session=sagemaker_session,
         script_mode=True,
         framework_version=tf_full_version,
-        py_version="py37",
+        py_version=py_version,
         metric_definitions=[{"Name": "train:global_steps", "Regex": r"global_step\/sec:\s(.*)"}],
         checkpoint_s3_uri=checkpoint_s3_uri,
         checkpoint_local_path=checkpoint_local_path,
@@ -139,7 +139,7 @@ def test_mnist_distributed(sagemaker_session, instance_type, tf_full_version, py
         train_instance_count=2,
         train_instance_type=instance_type,
         sagemaker_session=sagemaker_session,
-        py_version="py37",
+        py_version=py_version,
         script_mode=True,
         framework_version=tf_full_version,
         distributions=PARAMETER_SERVER_DISTRIBUTION,