diff --git a/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py b/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py index 5b0d5fd1fc..d54e586c14 100644 --- a/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py +++ b/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py @@ -130,6 +130,7 @@ def test_benchmark_workflow_end_to_end(): @pytest.mark.slow_test +@pytest.mark.gpu_intensive def test_recommendation_workflow_end_to_end(): """Run an AI recommendation via generate_deployment_recommendations and deploy the top recommendation.""" logger.info("Starting AI inference recommender recommendation integration test...") diff --git a/sagemaker-serve/tests/integ/test_jumpstart_integration.py b/sagemaker-serve/tests/integ/test_jumpstart_integration.py index 40532d2a37..e4e0f2725e 100644 --- a/sagemaker-serve/tests/integ/test_jumpstart_integration.py +++ b/sagemaker-serve/tests/integ/test_jumpstart_integration.py @@ -33,6 +33,7 @@ @pytest.mark.slow_test +@pytest.mark.gpu_intensive def test_jumpstart_build_deploy_invoke_cleanup(): """Integration test for JumpStart model build, deploy, invoke, and cleanup workflow""" logger.info("Starting JumpStart integration test...")