From 3fcbc27bf6ba2aa23158b1f2bf8604c9e40d299c Mon Sep 17 00:00:00 2001 From: Syed Jafri Date: Fri, 7 Aug 2026 13:12:19 -0700 Subject: [PATCH] fix: key check before logging training job name --- sagemaker-train/src/sagemaker/train/model_trainer.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/sagemaker-train/src/sagemaker/train/model_trainer.py b/sagemaker-train/src/sagemaker/train/model_trainer.py index 0b8ab91c2d..241b6ac3be 100644 --- a/sagemaker-train/src/sagemaker/train/model_trainer.py +++ b/sagemaker-train/src/sagemaker/train/model_trainer.py @@ -818,7 +818,9 @@ def train( """ training_request = self._create_training_job_args(input_data_config=input_data_config) - logger.info(f"Training Job Name: {training_request['training_job_name']}") + job_name = training_request.get("training_job_name") + if job_name: + logger.info(f"Training Job Name: {job_name}") if dry_run: logger.info("Dry-run validation passed. No job submitted.")