black format

Author: laurenyu

tests/integ/test_airflow_config.py

@@ -573,7 +573,7 @@ def test_xgboost_airflow_config_uploads_data_source_to_s3(
 
 @pytest.mark.canary_quick
 def test_pytorch_airflow_config_uploads_data_source_to_s3_when_inputs_not_provided(
-    sagemaker_session, cpu_instance_type, pytorch_full_version, pytorch_full_py_version,
+    sagemaker_session, cpu_instance_type, pytorch_full_version, pytorch_full_py_version
 ):
     with timeout(seconds=AIRFLOW_CONFIG_TIMEOUT_IN_SECONDS):
         estimator = PyTorch(

tests/integ/test_multidatamodel.py

@@ -318,12 +318,7 @@ def test_multi_data_model_deploy_trained_model_from_framework_estimator(
 
 
 def _mxnet_training_job(
-    sagemaker_session,
-    container_image,
-    mxnet_version,
-    py_version,
-    cpu_instance_type,
-    learning_rate,
+    sagemaker_session, container_image, mxnet_version, py_version, cpu_instance_type, learning_rate
 ):
     with timeout(minutes=TRAINING_DEFAULT_TIMEOUT_MINUTES):
         script_path = os.path.join(DATA_DIR, "mxnet_mnist", "mnist.py")