Skip to content

Commit 3fd7355

Browse files
author
Deng
committed
update tf py2 error unit test
1 parent 9b2c6d6 commit 3fd7355

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/unit/test_tf_estimator.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1020,7 +1020,7 @@ def test_py2_version_deprecated(sagemaker_session):
10201020
with pytest.raises(AttributeError) as e:
10211021
TensorFlow(
10221022
entry_point=SCRIPT_PATH,
1023-
framework_version="2.0.1",
1023+
framework_version="2.1.1",
10241024
role=ROLE,
10251025
sagemaker_session=sagemaker_session,
10261026
train_instance_count=INSTANCE_COUNT,
@@ -1029,7 +1029,7 @@ def test_py2_version_deprecated(sagemaker_session):
10291029
)
10301030

10311031
msg = (
1032-
"Python 2 containers are only available with 2.0.0 and lower versions. "
1032+
"Python 2 containers are only available with 2.1.0 and lower versions. "
10331033
"Please use a Python 3 container."
10341034
)
10351035
assert msg in str(e.value)

0 commit comments

Comments
 (0)