Reformatting pylint

Author: shantanutrip

src/sagemaker/estimator.py

@@ -3727,17 +3727,12 @@ def training_image_uri(self, region=None):
         Returns:
             str: The URI of the Docker image.
         """
-        # TODO: Revert
-        print(
-            f"""ABCD123 {getattr(self, "override_fw_version", None)}, self.framework_version:{self.framework_version}"""
-        )
-
         return image_uris.get_training_image_uri(
             region=region or self.sagemaker_session.boto_region_name,
             framework=self._framework_name,
             framework_version=getattr(
-                self, "override_fw_version", self.framework_version
-            ),  # pylint: disable=no-member
+                self, "override_fw_version", self.framework_version  # pylint: disable=no-member
+            ),
             py_version=self.py_version,  # pylint: disable=no-member
             image_uri=self.image_uri,
             distribution=getattr(self, "distribution", None),

src/sagemaker/tensorflow/estimator.py

@@ -182,14 +182,6 @@ def __init__(
             # TODO: Revert
             print(f"ABCD123 setting self.override_fw_version to {self.override_fw_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
-        # )
         self.py_version = py_version
         self.instance_type = instance_type