Skip to content

Commit 5eb482e

Browse files
chuyang-dengknakad
authored andcommitted
fix: update py2 warning message (aws#998)
Py2 support is determined by Py2 deprecation date, not TF version.
1 parent 6f3cf42 commit 5eb482e

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/tensorflow/estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ def _validate_args(
345345

346346
if py_version == "py2" and self._only_python_3_supported():
347347
msg = (
348-
"Python 2 containers are only available until TensorFlow version 1.14.0. "
348+
"Python 2 containers are only available until January 1st, 2020. "
349349
"Please use a Python 3 container."
350350
)
351351
raise AttributeError(msg)

tests/unit/test_tf_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -959,7 +959,7 @@ def test_py2_version_deprecated(sagemaker_session):
959959
with pytest.raises(AttributeError) as e:
960960
_build_tf(sagemaker_session=sagemaker_session, framework_version="1.14.1", py_version="py2")
961961

962-
msg = "Python 2 containers are only available until TensorFlow version 1.14.0. Please use a Python 3 container."
962+
msg = "Python 2 containers are only available until January 1st, 2020. Please use a Python 3 container."
963963
assert msg in str(e.value)
964964

965965

0 commit comments

Comments
 (0)