From 643330bf72f3b32954edaa72e7447d4b68bf3ecb Mon Sep 17 00:00:00 2001
From: EC2 Default User
 <ec2-user@ip-10-0-137-110.ap-northeast-1.compute.internal>
Date: Fri, 19 Jul 2024 01:54:02 +0000
Subject: [PATCH] Chore: fix C0301 - Line too long

---
 src/sagemaker/session.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/sagemaker/session.py b/src/sagemaker/session.py
index 2b1d6cc046..a438a1e257 100644
--- a/src/sagemaker/session.py
+++ b/src/sagemaker/session.py
@@ -8348,7 +8348,11 @@ def _logs_for_job(  # noqa: C901 - suppress complexity warning for this method
     """
     sagemaker_client = sagemaker_session.sagemaker_client
     request_end_time = time.time() + timeout if timeout else None
-    description = _wait_until(lambda: sagemaker_client.describe_training_job(TrainingJobName=job_name))
+    description = _wait_until(
+        lambda: sagemaker_client.describe_training_job(
+            TrainingJobName=job_name
+        )
+    )
     print(secondary_training_status_message(description, None), end="")
 
     instance_count, stream_names, positions, client, log_group, dot, color_wrap = _logs_init(