fix format

Author: ChoiByungWook

src/sagemaker/session.py

@@ -3514,12 +3514,14 @@ def get_caller_identity_arn(self):
             # Guessing this conditional's purpose was to handle lack of IAM permissions
             # https://github.com/aws/sagemaker-python-sdk/issues/2089#issuecomment-791802713
             if "AmazonSageMaker-ExecutionRole" in assumed_role:
-                LOGGER.warning('Assuming role was created in SageMaker AWS console, '
-                               'as the name contains `AmazonSageMaker-ExecutionRole`. '
-                               'Defaulting to Role ARN with service-role in path. '
-                               'If this Role ARN is incorrect, please add '
-                               'IAM read permissions to your role or supply the '
-                               'Role Arn directly.')
+                LOGGER.warning(
+                    "Assuming role was created in SageMaker AWS console, "
+                    "as the name contains `AmazonSageMaker-ExecutionRole`. "
+                    "Defaulting to Role ARN with service-role in path. "
+                    "If this Role ARN is incorrect, please add "
+                    "IAM read permissions to your role or supply the "
+                    "Role Arn directly."
+                )
                 role = re.sub(
                     r"^(.+)sts::(\d+):assumed-role/(.+?)/.*$",
                     r"\1iam::\2:role/service-role/\3",

tests/unit/test_session.py

@@ -406,15 +406,14 @@ def test_get_caller_identity_arn_from_an_execution_role(sts_regional_endpoint, b
     sess.boto_session.client("iam").get_role.return_value = {"Role": {"Arn": arn}}
 
     actual = sess.get_caller_identity_arn()
-    assert (
-        actual
-        == "arn:aws:iam::369233609183:role/AmazonSageMaker-ExecutionRole-20171129T072388"
-    )
+    assert actual == "arn:aws:iam::369233609183:role/AmazonSageMaker-ExecutionRole-20171129T072388"
 
 
 @patch("os.path.exists", side_effect=mock_exists(NOTEBOOK_METADATA_FILE, False))
 @patch("sagemaker.session.sts_regional_endpoint", return_value=STS_ENDPOINT)
-def test_get_caller_identity_arn_from_a_sagemaker_execution_role_with_iam_client_error(sts_regional_endpoint, boto_session):
+def test_get_caller_identity_arn_from_a_sagemaker_execution_role_with_iam_client_error(
+    sts_regional_endpoint, boto_session
+):
     sess = Session(boto_session)
     arn = "arn:aws:sts::369233609183:assumed-role/AmazonSageMaker-ExecutionRole-20171129T072388/SageMaker"
     sess.boto_session.client("sts", endpoint_url=STS_ENDPOINT).get_caller_identity.return_value = {