diff --git a/src/sagemaker/tensorflow/defaults.py b/src/sagemaker/tensorflow/defaults.py index b46e7fa195..be5439aaa0 100644 --- a/src/sagemaker/tensorflow/defaults.py +++ b/src/sagemaker/tensorflow/defaults.py @@ -13,4 +13,4 @@ """Placeholder docstring""" from __future__ import absolute_import -LATEST_PY2_VERSION = "2.1.0" +LATEST_PY2_VERSION = "2.1.1" diff --git a/src/sagemaker/tensorflow/estimator.py b/src/sagemaker/tensorflow/estimator.py index cb70ff1ef8..919e04a203 100644 --- a/src/sagemaker/tensorflow/estimator.py +++ b/src/sagemaker/tensorflow/estimator.py @@ -35,7 +35,7 @@ class TensorFlow(Framework): _framework_name = "tensorflow" _HIGHEST_LEGACY_MODE_ONLY_VERSION = version.Version("1.10.0") - _HIGHEST_PYTHON_2_VERSION = version.Version("2.1.0") + _HIGHEST_PYTHON_2_VERSION = version.Version("2.1.1") def __init__( self, diff --git a/tests/unit/sagemaker/tensorflow/test_estimator_init.py b/tests/unit/sagemaker/tensorflow/test_estimator_init.py index 2eb06388a9..ea82f20ccb 100644 --- a/tests/unit/sagemaker/tensorflow/test_estimator_init.py +++ b/tests/unit/sagemaker/tensorflow/test_estimator_init.py @@ -39,18 +39,18 @@ def _build_tf(sagemaker_session, **kwargs): @patch("sagemaker.fw_utils.python_deprecation_warning") def test_estimator_py2_deprecation_warning(warning, sagemaker_session): - estimator = _build_tf(sagemaker_session, framework_version="2.1.0", py_version="py2") + estimator = _build_tf(sagemaker_session, framework_version="2.1.1", py_version="py2") assert estimator.py_version == "py2" - warning.assert_called_with("tensorflow", "2.1.0") + warning.assert_called_with("tensorflow", "2.1.1") def test_py2_version_deprecated(sagemaker_session): with pytest.raises(AttributeError) as e: - _build_tf(sagemaker_session, framework_version="2.1.1", py_version="py2") + _build_tf(sagemaker_session, framework_version="2.1.2", py_version="py2") msg = ( - "Python 2 containers are only available with 2.1.0 and lower versions. " + "Python 2 containers are only available with 2.1.1 and lower versions. " "Please use a Python 3 container." ) assert msg in str(e.value)