style: Black-format two test files merged from master

test_load_or_create.py (core) and test_pipeline_class.py (mlops) landed
on master via aws#6256 after this branch's black pass and were pulled in
unformatted by the merge commit, failing black-check in CI.

---
X-AI-Prompt: fix failing codestyle-doc-tests reruns on PR aws#6294
X-AI-Tool: Kiro

Author: jam-jee

sagemaker-core/tests/unit/experiments/test_load_or_create.py

@@ -17,6 +17,7 @@
 already-exists branch against BOTH wordings so a future wording change cannot
 silently turn load-or-create back into a hard failure.
 """
+
 from __future__ import absolute_import
 
 import pytest
@@ -27,15 +28,12 @@
 from sagemaker.core.experiments.trial import _Trial
 from sagemaker.core.experiments.trial_component import _TrialComponent
 
-
 LEGACY_MESSAGE = "Experiment exp-1 already exists"
 NEW_MESSAGE = "Experiment names must be unique within an AWS account and region"
 
 
 def _validation_error(message):
-    return ClientError(
-        {"Error": {"Code": "ValidationException", "Message": message}}, "create"
-    )
+    return ClientError({"Error": {"Code": "ValidationException", "Message": message}}, "create")
 
 
 @pytest.mark.parametrize("message", [LEGACY_MESSAGE, NEW_MESSAGE])
@@ -74,9 +72,7 @@ def test_trial_load_or_create_loads_on_duplicate_name(message):
 def test_trial_component_load_or_create_loads_on_duplicate_name(message):
     with patch.object(_TrialComponent, "create", side_effect=_validation_error(message)):
         with patch.object(_TrialComponent, "load") as mock_load:
-            result, is_existed = _TrialComponent._load_or_create(
-                trial_component_name="tc-1"
-            )
+            result, is_existed = _TrialComponent._load_or_create(trial_component_name="tc-1")
 
     mock_load.assert_called_once_with("tc-1", None)
     assert result is mock_load.return_value

sagemaker-mlops/tests/unit/workflow/test_pipeline_class.py

@@ -422,9 +422,7 @@ def test_upsert_updates_existing_pipeline(self, mock_session):
                             # Verify tags were merged and added
                             mock_session.sagemaker_client.add_tags.assert_called_once()
 
-    def test_upsert_updates_existing_pipeline_with_names_must_be_unique_message(
-        self, mock_session
-    ):
+    def test_upsert_updates_existing_pipeline_with_names_must_be_unique_message(self, mock_session):
         """Upsert must also recognize the newer service wording 'names must be unique'."""
         error_response = {
             "Error": {
@@ -436,11 +434,15 @@ def test_upsert_updates_existing_pipeline_with_names_must_be_unique_message(
         mock_session.sagemaker_client.list_tags = Mock(return_value={"Tags": []})
         mock_session.sagemaker_client.add_tags = Mock()
 
-        with patch.object(Pipeline, 'create') as mock_create:
-            with patch.object(Pipeline, 'update') as mock_update:
-                with patch('sagemaker.mlops.workflow.pipeline.resolve_value_from_config') as mock_resolve:
-                    with patch('sagemaker.mlops.workflow.pipeline.resolve_and_validate_role') as mock_validate:
-                        with patch('sagemaker.mlops.workflow.pipeline.format_tags') as mock_format:
+        with patch.object(Pipeline, "create") as mock_create:
+            with patch.object(Pipeline, "update") as mock_update:
+                with patch(
+                    "sagemaker.mlops.workflow.pipeline.resolve_value_from_config"
+                ) as mock_resolve:
+                    with patch(
+                        "sagemaker.mlops.workflow.pipeline.resolve_and_validate_role"
+                    ) as mock_validate:
+                        with patch("sagemaker.mlops.workflow.pipeline.format_tags") as mock_format:
                             mock_resolve.return_value = "arn:aws:iam::123:role/SageMakerRole"
                             mock_validate.return_value = "arn:aws:iam::123:role/SageMakerRole"
                             mock_format.return_value = []
@@ -450,8 +452,7 @@ def test_upsert_updates_existing_pipeline_with_names_must_be_unique_message(
                             }
 
                             pipeline = Pipeline(
-                                name="test-pipeline",
-                                sagemaker_session=mock_session
+                                name="test-pipeline", sagemaker_session=mock_session
                             )
 
                             result = pipeline.upsert(role_arn="arn:aws:iam::123:role/SageMakerRole")
@@ -475,11 +476,15 @@ def test_upsert_of_existing_pipeline_does_not_log_error_hint(self, mock_session,
         )
         mock_session.sagemaker_client.list_tags = Mock(return_value={"Tags": []})
 
-        with patch.object(Pipeline, 'update') as mock_update:
-            with patch.object(Pipeline, '_create_args') as mock_args:
-                with patch('sagemaker.mlops.workflow.pipeline.resolve_value_from_config') as mock_resolve:
-                    with patch('sagemaker.mlops.workflow.pipeline.resolve_and_validate_role') as mock_validate:
-                        with patch('sagemaker.mlops.workflow.pipeline.format_tags') as mock_format:
+        with patch.object(Pipeline, "update") as mock_update:
+            with patch.object(Pipeline, "_create_args") as mock_args:
+                with patch(
+                    "sagemaker.mlops.workflow.pipeline.resolve_value_from_config"
+                ) as mock_resolve:
+                    with patch(
+                        "sagemaker.mlops.workflow.pipeline.resolve_and_validate_role"
+                    ) as mock_validate:
+                        with patch("sagemaker.mlops.workflow.pipeline.format_tags") as mock_format:
                             mock_resolve.return_value = "arn:aws:iam::123:role/SageMakerRole"
                             mock_validate.return_value = "arn:aws:iam::123:role/SageMakerRole"
                             mock_format.return_value = []
@@ -489,12 +494,13 @@ def test_upsert_of_existing_pipeline_does_not_log_error_hint(self, mock_session,
                             }
 
                             pipeline = Pipeline(
-                                name="test-pipeline",
-                                sagemaker_session=mock_session
+                                name="test-pipeline", sagemaker_session=mock_session
                             )
 
                             with caplog.at_level(logging.ERROR):
-                                result = pipeline.upsert(role_arn="arn:aws:iam::123:role/SageMakerRole")
+                                result = pipeline.upsert(
+                                    role_arn="arn:aws:iam::123:role/SageMakerRole"
+                                )
 
         assert "PipelineArn" in result
         mock_update.assert_called_once()
@@ -515,19 +521,20 @@ def test_bare_create_of_existing_pipeline_logs_error_hint(self, mock_session, ca
             side_effect=ClientError(error_response, "create_pipeline")
         )
 
-        with patch.object(Pipeline, '_create_args') as mock_args:
-            with patch('sagemaker.mlops.workflow.pipeline.resolve_value_from_config') as mock_resolve:
-                with patch('sagemaker.mlops.workflow.pipeline.resolve_and_validate_role') as mock_validate:
-                    with patch('sagemaker.mlops.workflow.pipeline.format_tags') as mock_format:
+        with patch.object(Pipeline, "_create_args") as mock_args:
+            with patch(
+                "sagemaker.mlops.workflow.pipeline.resolve_value_from_config"
+            ) as mock_resolve:
+                with patch(
+                    "sagemaker.mlops.workflow.pipeline.resolve_and_validate_role"
+                ) as mock_validate:
+                    with patch("sagemaker.mlops.workflow.pipeline.format_tags") as mock_format:
                         mock_resolve.return_value = "arn:aws:iam::123:role/SageMakerRole"
                         mock_validate.return_value = "arn:aws:iam::123:role/SageMakerRole"
                         mock_format.return_value = []
                         mock_args.return_value = {"PipelineName": "test-pipeline"}
 
-                        pipeline = Pipeline(
-                            name="test-pipeline",
-                            sagemaker_session=mock_session
-                        )
+                        pipeline = Pipeline(name="test-pipeline", sagemaker_session=mock_session)
 
                         with caplog.at_level(logging.ERROR):
                             with pytest.raises(ClientError):