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.""" 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()