fix codestyle 2

Author: mollyheamazon

tests/integ/sagemaker/experiments/test_run.py

@@ -86,9 +86,7 @@ def verify_load_run():
         retry_with_backoff(verify_load_run, 4)
 
 
-def test_two_local_run_init_with_same_run_name_and_different_exp_names(
-    sagemaker_session
-):
+def test_two_local_run_init_with_same_run_name_and_different_exp_names(sagemaker_session):
     exp_name1 = f"my-two-local-exp1-{name()}"
     exp_name2 = f"my-two-local-exp2-{name()}"
     run_name = "test-run"
@@ -126,9 +124,7 @@ def test_two_local_run_init_with_same_run_name_and_different_exp_names(
         ("my-test4", "test-run", "run-display-name-test"),  # with supplied display name
     ],
 )
-def test_run_name_vs_trial_component_name_edge_cases(
-    sagemaker_session, input_names
-):
+def test_run_name_vs_trial_component_name_edge_cases(sagemaker_session, input_names):
     exp_name, run_name, run_display_name = input_names
     with cleanup_exp_resources(exp_names=[exp_name], sagemaker_session=sagemaker_session):
         with Run(

tests/integ/test_feature_store.py

@@ -1646,8 +1646,8 @@ def test_create_dataset_with_feature_group_base(
         )
 
         with (
-            timeout(minutes=10) 
-            and cleanup_offline_store(base, feature_store_session) 
+            timeout(minutes=10)
+            and cleanup_offline_store(base, feature_store_session)
             and cleanup_offline_store(feature_group, feature_store_session)
         ):
             feature_store = FeatureStore(sagemaker_session=feature_store_session)
@@ -1835,8 +1835,8 @@ def test_create_dataset_with_feature_group_base_with_additional_params(
         )
 
         with (
-            timeout(minutes=10) 
-            and cleanup_offline_store(base, feature_store_session) 
+            timeout(minutes=10)
+            and cleanup_offline_store(base, feature_store_session)
             and cleanup_offline_store(feature_group, feature_store_session)
         ):
             feature_store = FeatureStore(sagemaker_session=feature_store_session)

tests/unit/sagemaker/modules/train/test_model_trainer.py

@@ -1061,7 +1061,7 @@ def mock_upload_data(path, bucket, key_prefix):
         input_data_config=ANY,
         hyper_parameters=hyperparameters,
         environment=environment,
-)
+    )
 
 
 def test_safe_configs():