Fixing py2 support for latest TF version

Author: Sam Oshin

tests/unit/test_tf_estimator.py

@@ -991,6 +991,13 @@ def test_py2_version_is_not_deprecated(sagemaker_session):
     estimator = _build_tf(sagemaker_session=sagemaker_session, framework_version="1.15.0", py_version="py2")
     assert estimator.py_version == "py2"
 
+def test_py2_version_is_not_deprecated(sagemaker_session):
+    estimator = _build_tf(
+        sagemaker_session=sagemaker_session, framework_version="1.15.0", py_version="py2"
+    )
+    assert estimator.py_version == "py2"
+
+
 def test_py3_is_default_version_before_tf1_14(sagemaker_session):
     estimator = _build_tf(sagemaker_session=sagemaker_session, framework_version="1.13")