From 8ed55771b1a65759c7dd1d2e443f4629e90de709 Mon Sep 17 00:00:00 2001 From: Roja Reddy Sareddy Date: Wed, 12 Aug 2026 15:58:43 -0700 Subject: [PATCH 1/2] fix: update timeout of RLVR sequence_length test --- .../tests/integ/train/test_rlvr_trainer_integration.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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..fc28cda169 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 = 7200 # 2 hour timeout (sequence_length training takes >1 hour) poll_interval = 30 start_time = time.time() From a2ccf6779e73201b3ddd5440bbf897149bef43c8 Mon Sep 17 00:00:00 2001 From: Roja Reddy Sareddy Date: Wed, 12 Aug 2026 23:18:59 -0700 Subject: [PATCH 2/2] fix: add xdist_group to Nova Bedrock deployment tests --- .../tests/integ/test_nova_model_customization_deployment.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sagemaker-serve/tests/integ/test_nova_model_customization_deployment.py b/sagemaker-serve/tests/integ/test_nova_model_customization_deployment.py index d8b0a2b787..65f57ff12d 100644 --- a/sagemaker-serve/tests/integ/test_nova_model_customization_deployment.py +++ b/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."""