Fixed formatting issues

Author: Ameen Khan

src/sagemaker/workflow/pipeline.py

@@ -121,10 +121,7 @@ def create(
         return self.sagemaker_session.sagemaker_client.create_pipeline(**kwargs)
 
     def _create_args(
-        self,
-        role_arn: str,
-        description: str,
-        parallelism_config: ParallelismConfiguration
+        self, role_arn: str, description: str, parallelism_config: ParallelismConfiguration
     ):
         """Constructs the keyword argument dict for a create_pipeline call.
 
@@ -163,9 +160,7 @@ def _create_args(
             }
 
         update_args(
-            kwargs,
-            PipelineDescription=description,
-            ParallelismConfiguration=parallelism_config
+            kwargs, PipelineDescription=description, ParallelismConfiguration=parallelism_config
         )
         return kwargs
 

tests/integ/test_workflow.py

@@ -72,7 +72,6 @@
 )
 from sagemaker.workflow.step_collections import RegisterModel
 from sagemaker.workflow.pipeline import Pipeline
-from sagemaker.workflow.parallelism_config import ParallelismConfiguration
 from sagemaker.lambda_helper import Lambda
 from sagemaker.feature_store.feature_group import FeatureGroup, FeatureDefinition, FeatureTypeEnum
 from tests.integ import DATA_DIR
@@ -2303,7 +2302,8 @@ def test_large_pipeline(sagemaker_session, role, pipeline_name, region_name):
             sqs_queue_url="https://sqs.us-east-2.amazonaws.com/123456789012/MyQueue",
             inputs={"arg1": "foo"},
             outputs=[outputParam],
-        ) for count in range(2000)
+        )
+        for count in range(2000)
     ]
     pipeline = Pipeline(
         name=pipeline_name,
@@ -2335,7 +2335,10 @@ def test_large_pipeline(sagemaker_session, role, pipeline_name, region_name):
         except Exception:
             pass
 
-def test_create_and_update_with_parallelism_config(sagemaker_session, role, pipeline_name, region_name):
+
+def test_create_and_update_with_parallelism_config(
+    sagemaker_session, role, pipeline_name, region_name
+):
     instance_count = ParameterInteger(name="InstanceCount", default_value=2)
 
     outputParam = CallbackOutput(output_name="output", output_type=CallbackOutputTypeEnum.String)
@@ -2356,7 +2359,6 @@ def test_create_and_update_with_parallelism_config(sagemaker_session, role, pipe
         sagemaker_session=sagemaker_session,
     )
 
-
     try:
         response = pipeline.create(role, parallelism_config={"MaxParallelExecutionSteps": 50})
         create_arn = response["PipelineArn"]