Add Print

Author: shantanutrip

src/sagemaker/estimator.py

@@ -3728,6 +3728,8 @@ def training_image_uri(self, region=None):
         Returns:
             str: The URI of the Docker image.
         """
+        ## TODO: Revert
+        print(f"ABCD1234 self.override_fw_version: {self.override_fw_version} and self.framework_version: {self.framework_version}")
 
         return image_uris.get_training_image_uri(
             region=region or self.sagemaker_session.boto_region_name,

src/sagemaker/tensorflow/estimator.py

@@ -178,6 +178,8 @@ def __init__(
             )
         self.framework_version = framework_version
         self.override_fw_version = "2.16" if self.framework_version and "2.16" in self.framework_version else None
+        ## 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.