Revert "debug: adding additonal logs to investigate tests"

This reverts commit f71d183.

Author: Syed Jafri

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

@@ -276,21 +276,10 @@ 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(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}")
+            logger.error("Error: %s", e)
             raise e
 
         if wait:

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

@@ -270,20 +270,10 @@ 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(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}")
+            logger.error("Error: %s", e)
             raise e
 
         if wait: