fix: black check failure

Author: EthanShouhanCheng

src/sagemaker/workflow/pipeline.py

@@ -184,17 +184,16 @@ def upsert(
                 response = self.update(role_arn, description)
                 if tags is not None:
                     old_tags = self.sagemaker_session.sagemaker_client.list_tags(
-                        ResourceArn=response["PipelineArn"])["Tags"]
+                        ResourceArn=response["PipelineArn"]
+                    )["Tags"]
 
                     tag_keys = [tag["Key"] for tag in old_tags]
 
                     self.sagemaker_session.sagemaker_client.delete_tags(
-                        ResourceArn=response["PipelineArn"],
-                        TagKeys=tag_keys
+                        ResourceArn=response["PipelineArn"], TagKeys=tag_keys
                     )
                     self.sagemaker_session.sagemaker_client.add_tags(
-                        ResourceArn=response["PipelineArn"],
-                        Tags=tags
+                        ResourceArn=response["PipelineArn"], Tags=tags
                     )
             else:
                 raise

tests/integ/test_workflow.py

@@ -458,7 +458,9 @@ def test_one_step_sklearn_processing_pipeline(
         # sagemaker entities. However, the jobs created in the steps themselves execute
         # under a potentially different role, often requiring access to S3 and other
         # artifacts not required to during creation of the jobs in the pipeline steps.
-        response = pipeline.create(role, tags=[{"Key": "foo", "Value": "abc"}, {"Key": "bar", "Value": "xyz"}])
+        response = pipeline.create(
+            role, tags=[{"Key": "foo", "Value": "abc"}, {"Key": "bar", "Value": "xyz"}]
+        )
         create_arn = response["PipelineArn"]
         assert re.match(
             fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",
@@ -469,7 +471,9 @@ def test_one_step_sklearn_processing_pipeline(
             assert tag in original_tags["Tags"]
 
         pipeline.parameters = [ParameterInteger(name="InstanceCount", default_value=1)]
-        response = pipeline.upsert(role, tags=[{"Key": "abc", "Value": "foo"}, {"Key": "xyz", "Value": "bar"}])
+        response = pipeline.upsert(
+            role, tags=[{"Key": "abc", "Value": "foo"}, {"Key": "xyz", "Value": "bar"}]
+        )
         update_arn = response["PipelineArn"]
         assert re.match(
             fr"arn:aws:sagemaker:{region_name}:\d{{12}}:pipeline/{pipeline_name}",

tests/unit/sagemaker/workflow/test_pipeline.py

@@ -103,9 +103,7 @@ def test_pipeline_upsert(sagemaker_session_mock, role_arn):
             },
         ),
         {"PipelineArn": "mock_pipeline_arn"},
-        [
-            {"Key": "dummy", "Value": "dummy_tag"}
-        ],
+        [{"Key": "dummy", "Value": "dummy_tag"}],
         {},
         {},
     ]
@@ -128,7 +126,9 @@ def test_pipeline_upsert(sagemaker_session_mock, role_arn):
     assert sagemaker_session_mock.sagemaker_client.update_pipeline.called_with(
         PipelineName="MyPipeline", PipelineDefinition=pipeline.definition(), RoleArn=role_arn
     )
-    assert sagemaker_session_mock.sagemaker_client.list_tags.called_with(ResourceArn="mock_pipeline_arn")
+    assert sagemaker_session_mock.sagemaker_client.list_tags.called_with(
+        ResourceArn="mock_pipeline_arn"
+    )
     assert sagemaker_session_mock.sagemaker_client.delete_tags(
         ResourceArn="mock_pipeline_arn", TagKeys=["dummy"]
     )