diff --git a/sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py b/sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py index a022b6846f..bf2b348ddd 100644 --- a/sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py +++ b/sagemaker-train/tests/integ/train/test_rlvr_trainer_integration.py @@ -376,7 +376,7 @@ def test_rlvr_trainer_lora_with_sequence_length(sagemaker_session): training_job = rlvr_trainer.train(wait=False) logger.info(f"Training job submitted: {training_job.training_job_arn}") - max_wait_time = 3600 + max_wait_time = 10800 # 3h — GRPO RLVR on a 9B model at 8K is compute-heavy poll_interval = 30 start_time = time.time()