fix: update timeout of RLVR sequence_length test (#6180)

* fix: update timeout of RLVR sequence_length test

* fix: add xdist_group to Nova Bedrock deployment tests

---------

Co-authored-by: Roja Reddy Sareddy <rsareddy@amazon.com>

Author: rsareddy0329

sagemaker-serve/tests/integ/test_nova_model_customization_deployment.py

@@ -469,6 +469,7 @@ def test_rlvr_trainer_build(self, training_job_name, sagemaker_session):
 
 
 @pytest.mark.us_east_1
+@pytest.mark.xdist_group("bedrock_deployment")
 class TestNovaBedrockDeployment:
     """Test deploying a fine-tuned Nova model to Amazon Bedrock as a custom model."""
 

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 = 7200  # 2 hour timeout (sequence_length training takes >1 hour)
     poll_interval = 30
     start_time = time.time()