Skip to content

Commit 158e7e1

Browse files
authored
test: move two tests in serve to gpu-integ-tests (#6096)
1 parent a76fb88 commit 158e7e1

2 files changed

Lines changed: 2 additions & 0 deletions

File tree

‎sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,7 @@ def test_benchmark_workflow_end_to_end():
130130

131131

132132
@pytest.mark.slow_test
133+
@pytest.mark.gpu_intensive
133134
def test_recommendation_workflow_end_to_end():
134135
"""Run an AI recommendation via generate_deployment_recommendations and deploy the top recommendation."""
135136
logger.info("Starting AI inference recommender recommendation integration test...")

‎sagemaker-serve/tests/integ/test_jumpstart_integration.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333

3434

3535
@pytest.mark.slow_test
36+
@pytest.mark.gpu_intensive
3637
def test_jumpstart_build_deploy_invoke_cleanup():
3738
"""Integration test for JumpStart model build, deploy, invoke, and cleanup workflow"""
3839
logger.info("Starting JumpStart integration test...")

0 commit comments

Comments
 (0)