Reformatting

Author: shantanutrip

src/sagemaker/estimator.py

@@ -3727,13 +3727,17 @@ def training_image_uri(self, region=None):
         Returns:
             str: The URI of the Docker image.
         """
-        ## TODO: Revert
-        print(f"""ABCD1234 self.override_fw_version: {getattr(self, "override_fw_version", None)} and self.framework_version: {self.framework_version}""")
+        # 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
+            framework_version=getattr(
+                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

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

tests/integ/sagemaker/workflow/test_training_steps.py

@@ -243,7 +243,7 @@ def test_tensorflow_training_step_with_parameterized_code_input(
             "This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
             "https://github.com/tensorflow/io/issues/2039"
         )
-    
+
     base_dir = os.path.join(DATA_DIR, "tensorflow_mnist")
     entry_point1 = "mnist_v2.py"
     entry_point2 = "mnist_dummy.py"

tests/integ/test_transformer.py

@@ -561,7 +561,7 @@ def test_transform_tf_kms_network_isolation(
             "This test is failing in TensorFlow 2.16 beacuse of an upstream bug: "
             "https://github.com/tensorflow/io/issues/2039"
         )
-    
+
     data_path = os.path.join(DATA_DIR, "tensorflow_mnist")
 
     tf = TensorFlow(

tests/unit/test_processing.py

@@ -509,7 +509,14 @@ def test_tensorflow_processor_with_required_parameters(
     else:
         tensorflow_image_uri = (
             "763104351884.dkr.ecr.us-west-2.amazonaws.com/tensorflow-training:{}-cpu-{}"
-        ).format("2.16" if Version(tensorflow_training_version) in SpecifierSet("==2.16.*") else tensorflow_training_version, tensorflow_training_py_version)
+        ).format(
+            (
+                "2.16"
+                if Version(tensorflow_training_version) in SpecifierSet("==2.16.*")
+                else tensorflow_training_version
+            ),
+            tensorflow_training_py_version,
+        )
 
     expected_args["app_specification"]["ImageUri"] = tensorflow_image_uri