Skip to content

Commit f209e6e

Browse files
committed
fix: use correct TF version in empty framework_version warning
1 parent 2763f9a commit f209e6e

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
@@ -229,7 +229,7 @@ def __init__(self, training_steps=None, evaluation_steps=None, checkpoint_path=N
229229
**kwargs: Additional kwargs passed to the Framework constructor.
230230
"""
231231
if framework_version is None:
232-
LOGGER.warning(fw.empty_framework_version_warning(TF_VERSION, TF_VERSION))
232+
LOGGER.warning(fw.empty_framework_version_warning(TF_VERSION, self.LATEST_VERSION))
233233
self.framework_version = framework_version or TF_VERSION
234234

235235
super(TensorFlow, self).__init__(image_name=image_name, **kwargs)

tests/unit/test_tf_estimator.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -697,7 +697,7 @@ def test_empty_framework_version(warning, sagemaker_session):
697697
framework_version=None)
698698

699699
assert estimator.framework_version == defaults.TF_VERSION
700-
warning.assert_called_with(defaults.TF_VERSION, defaults.TF_VERSION)
700+
warning.assert_called_with(defaults.TF_VERSION, estimator.LATEST_VERSION)
701701

702702

703703
def _deprecated_args_msg(args):

0 commit comments

Comments
 (0)