running black-format

Author: Shikha Panghal

src/sagemaker/estimator.py

@@ -1125,8 +1125,9 @@ def _prepare_init_params_from_job_description(cls, job_details, model_channel_na
                 init_params["max_wait"] = max_wait
 
         if job_details.get("RetryStrategy", False):
-            init_params["max_retry_attempts"] = job_details.get("RetryStrategy", {})\
-                .get("MaximumRetryAttempts")
+            init_params["max_retry_attempts"] = job_details.get("RetryStrategy", {}).get(
+                "MaximumRetryAttempts"
+            )
             max_wait = job_details.get("StoppingCondition", {}).get("MaxWaitTimeInSeconds")
             if max_wait:
                 init_params["max_wait"] = max_wait

tests/integ/test_tf.py

@@ -95,11 +95,9 @@ def test_mnist_with_checkpoint_config(
     expected_retry_strategy = {
         "MaximumRetryAttempts": 2,
     }
-    actual_retry_strategy = (
-        sagemaker_session.sagemaker_client.describe_training_job(TrainingJobName=training_job_name)[
-            "RetryStrategy"
-        ]
-    )
+    actual_retry_strategy = sagemaker_session.sagemaker_client.describe_training_job(
+        TrainingJobName=training_job_name
+    )["RetryStrategy"]
     assert actual_training_checkpoint_config == expected_training_checkpoint_config
     assert actual_training_environment_variable_config == ENV_INPUT
     assert actual_retry_strategy == expected_retry_strategy

tests/unit/test_estimator.py

@@ -3183,9 +3183,7 @@ def test_prepare_init_params_from_job_description_with_spot_training():
 
 def test_prepare_init_params_from_job_description_with_retry_strategy():
     job_description = RETURNED_JOB_DESCRIPTION.copy()
-    job_description["RetryStrategy"] = {
-        "MaximumRetryAttempts": 2
-    }
+    job_description["RetryStrategy"] = {"MaximumRetryAttempts": 2}
     job_description["StoppingCondition"] = {
         "MaxRuntimeInSeconds": 86400,
         "MaxWaitTimeInSeconds": 87000,