fix: use correct TF version in empty framework_version warning (#756)

Author: laurenyu

src/sagemaker/tensorflow/estimator.py

@@ -228,7 +228,7 @@ def __init__(self, training_steps=None, evaluation_steps=None, checkpoint_path=N
             **kwargs: Additional kwargs passed to the Framework constructor.
         """
         if framework_version is None:
-            LOGGER.warning(fw.empty_framework_version_warning(TF_VERSION, TF_VERSION))
+            LOGGER.warning(fw.empty_framework_version_warning(TF_VERSION, self.LATEST_VERSION))
         self.framework_version = framework_version or TF_VERSION
 
         super(TensorFlow, self).__init__(image_name=image_name, **kwargs)

tests/unit/test_tf_estimator.py

@@ -697,7 +697,7 @@ def test_empty_framework_version(warning, sagemaker_session):
                            framework_version=None)
 
     assert estimator.framework_version == defaults.TF_VERSION
-    warning.assert_called_with(defaults.TF_VERSION, defaults.TF_VERSION)
+    warning.assert_called_with(defaults.TF_VERSION, estimator.LATEST_VERSION)
 
 
 def _deprecated_args_msg(args):