fixing flake8 build

Author: Shikha Panghal

src/sagemaker/estimator.py

@@ -1130,8 +1130,6 @@ def _prepare_init_params_from_job_description(cls, job_details, model_channel_na
             max_wait = job_details.get("StoppingCondition", {}).get("MaxWaitTimeInSeconds")
             if max_wait:
                 init_params["max_wait"] = max_wait
-
-
         return init_params
 
     def transformer(
@@ -1545,7 +1543,6 @@ def _get_train_args(cls, estimator, inputs, experiment_config):
 
         return train_args
 
-
     @classmethod
     def _add_spot_checkpoint_args(cls, local_mode, estimator, train_args):
         """Placeholder docstring"""

tests/integ/test_tf.py

@@ -105,7 +105,6 @@ def test_mnist_with_checkpoint_config(
     assert actual_retry_strategy == expected_retry_strategy
 
 
-
 def test_server_side_encryption(sagemaker_session, tf_full_version, tf_full_py_version):
     with kms_utils.bucket_with_encryption(sagemaker_session, ROLE) as (bucket_with_kms, kms_key):
         output_path = os.path.join(

tests/unit/test_estimator.py

@@ -1031,6 +1031,7 @@ def test_training_job_with_rule_job_summary(sagemaker_session, training_job_desc
         },
     ]
 
+
 def test_framework_with_spot_and_checkpoints(sagemaker_session):
     f = DummyFramework(
         "my_script.py",

tests/unit/test_mxnet.py

@@ -994,6 +994,7 @@ def test_mx_missing_environment_variables(
     )
     assert not mx.environment
 
+
 def test_mx_enable_sm_metrics(sagemaker_session, mxnet_training_version, mxnet_training_py_version):
     mx = MXNet(
         entry_point=SCRIPT_PATH,