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.")