Comment previous data

Author: shantanutrip

src/sagemaker/tensorflow/estimator.py

@@ -176,13 +176,15 @@ def __init__(
             logger.warning(
                 fw.python_deprecation_warning(self._framework_name, defaults.LATEST_PY2_VERSION)
             )
+        self.framework_version = framework_version
         # TF training and inference versions do not have a one-to-one connection. This mismatch
         # is accommodated by the underlying dictionary. The key of dictionary relates to the inference
         # version and the value relates to training version.
-        training_inference_version_mismatch_dict = {"2.16.1": "2.16.2"}
-        self.framework_version = training_inference_version_mismatch_dict.get(
-            framework_version, framework_version
-        )
+
+        # training_inference_version_mismatch_dict = {"2.16.1": "2.16.2"}
+        # self.framework_version = training_inference_version_mismatch_dict.get(
+        #     framework_version, framework_version
+        # )
         self.py_version = py_version
         self.instance_type = instance_type