debug: adding additonal logs to investigate tests

Author: Syed Jafri

sagemaker-train/src/sagemaker/train/rlvr_trainer.py

@@ -276,10 +276,21 @@ def train(self, training_dataset: Optional[Union[str, DataSet]] = None,
         if self.stopping_condition is not None:
             create_args["stopping_condition"] = self.stopping_condition
 
+        # Log the IAM role being used
+        logger.info(f"IAM Role ARN: {role}")
+
+        # Log the full training job arguments for debugging
+        logger.info(f"TrainingJob.create() arguments: {create_args}")
+
         try:
             training_job = TrainingJob.create(**create_args)
         except Exception as e:
-            logger.error("Error: %s", e)
+            logger.error(f"Error creating training job: {e}")
+            logger.error(f"Training job name: {current_training_job_name}")
+            logger.error(f"Serverless config: {serverless_config}")
+            logger.error(f"Evaluator ARN: {evaluator_arn}")
+            logger.error(f"Role ARN: {role}")
+            logger.error(f"Full create_args: {create_args}")
             raise e
 
         if wait:

sagemaker-train/src/sagemaker/train/sft_trainer.py

@@ -270,10 +270,20 @@ def train(self, training_dataset: Optional[Union[str, DataSet]] = None, validati
         if self.stopping_condition is not None:
             create_args["stopping_condition"] = self.stopping_condition
 
+        # Log the IAM role being used
+        logger.info(f"IAM Role ARN: {role}")
+
+        # Log the full training job arguments for debugging
+        logger.info(f"TrainingJob.create() arguments: {create_args}")
+
         try:
             training_job = TrainingJob.create(**create_args)
         except Exception as e:
-            logger.error("Error: %s", e)
+            logger.error(f"Error creating training job: {e}")
+            logger.error(f"Training job name: {current_training_job_name}")
+            logger.error(f"Serverless config: {serverless_config}")
+            logger.error(f"Role ARN: {role}")
+            logger.error(f"Full create_args: {create_args}")
             raise e
 
         if wait: