From 91bdacd7e3fe5f4939ad3d37985048f0cbba8b88 Mon Sep 17 00:00:00 2001 From: Syed Jafri Date: Wed, 22 Jul 2026 17:31:51 -0700 Subject: [PATCH] fix: stream_logs_smhp extract training job from obj --- sagemaker-train/src/sagemaker/train/base_trainer.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/sagemaker-train/src/sagemaker/train/base_trainer.py b/sagemaker-train/src/sagemaker/train/base_trainer.py index d3c5957c12..468dacd610 100644 --- a/sagemaker-train/src/sagemaker/train/base_trainer.py +++ b/sagemaker-train/src/sagemaker/train/base_trainer.py @@ -701,7 +701,12 @@ def _stream_logs_smtj(self, training_job, poll: int) -> None: def _stream_logs_smhp(self, training_job, compute, poll: int, start_time_ms=None) -> None: """Stream logs for a HyperPod job using filter_log_events polling.""" - job_id = training_job if isinstance(training_job, str) else str(training_job) + if isinstance(training_job, str): + job_id = training_job + elif hasattr(training_job, 'training_job_name'): + job_id = training_job.training_job_name + else: + job_id = str(training_job) sagemaker_session = TrainDefaults.get_sagemaker_session( sagemaker_session=self.sagemaker_session