From c4c1c7ffefb1ee54440adccdfbaeefdd42af79a6 Mon Sep 17 00:00:00 2001
From: Lucas Jia <lucasjia@amazon.com>
Date: Tue, 18 Aug 2026 15:41:08 -0700
Subject: [PATCH] test(serve): Add skip_in_pr_check marker for hang-prone integ
 tests

---
 sagemaker-serve/pyproject.toml                                 | 3 +++
 .../tests/integ/test_ai_inference_recommender_integration.py   | 1 +
 .../tests/integ/test_model_customization_deployment.py         | 1 +
 sagemaker-serve/tests/integ/test_optimize_integration.py       | 1 +
 sagemaker-serve/tox.ini                                        | 1 +
 5 files changed, 7 insertions(+)

diff --git a/sagemaker-serve/pyproject.toml b/sagemaker-serve/pyproject.toml
index 618a42ca32..27ffe47ab8 100644
--- a/sagemaker-serve/pyproject.toml
+++ b/sagemaker-serve/pyproject.toml
@@ -73,6 +73,9 @@ python_files = ["test_*.py"]
 python_classes = ["Test*"]
 python_functions = ["test_*"]
 addopts = "-v --tb=short"
+markers = [
+    "skip_in_pr_check: mark a test that is excluded from PR check runs. Long-running or hang-prone tests that would otherwise push the run past the CodeBuild timeout; they run in a dedicated scheduled CI run instead.",
+]
 
 [tool.black]
 line-length = 100
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 d54e586c14..634eef014a 100644
--- a/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py
+++ b/sagemaker-serve/tests/integ/test_ai_inference_recommender_integration.py
@@ -70,6 +70,7 @@ def _build_jumpstart_model_builder(role_arn):
 
 
 @pytest.mark.slow_test
+@pytest.mark.skip_in_pr_check
 def test_benchmark_workflow_end_to_end():
     """Deploy a JumpStart endpoint, run a benchmark against it, parse the result."""
     logger.info("Starting AI inference recommender benchmark integration test...")
diff --git a/sagemaker-serve/tests/integ/test_model_customization_deployment.py b/sagemaker-serve/tests/integ/test_model_customization_deployment.py
index 63bf3f3811..694273eb70 100644
--- a/sagemaker-serve/tests/integ/test_model_customization_deployment.py
+++ b/sagemaker-serve/tests/integ/test_model_customization_deployment.py
@@ -107,6 +107,7 @@ def test_build_from_training_job(self, training_job_name, sagemaker_session):
         assert model_builder.image_uri is not None
         assert model_builder.instance_type is not None
 
+    @pytest.mark.skip_in_pr_check
     def test_deploy_from_training_job(self, training_job_name, endpoint_name, cleanup_endpoints, sagemaker_session):
         """Test deploying model from training job.
 
diff --git a/sagemaker-serve/tests/integ/test_optimize_integration.py b/sagemaker-serve/tests/integ/test_optimize_integration.py
index 9ee222ea9e..0f28f5bafd 100644
--- a/sagemaker-serve/tests/integ/test_optimize_integration.py
+++ b/sagemaker-serve/tests/integ/test_optimize_integration.py
@@ -39,6 +39,7 @@
 DJL_LMI_VERSION = "0.31.0"
 
 
+@pytest.mark.skip_in_pr_check
 def test_optimize_build_deploy_invoke_cleanup():
     """Integration test for Optimize workflow"""
     logger.info("Starting Optimize integration test...")
diff --git a/sagemaker-serve/tox.ini b/sagemaker-serve/tox.ini
index f13299b96e..d5e8b110ec 100644
--- a/sagemaker-serve/tox.ini
+++ b/sagemaker-serve/tox.ini
@@ -66,6 +66,7 @@ markers =
     gpu_intensive: mark a test as GPU resource intensive (runs on scheduled CI, not PR checks).
     us_east_1: mark a test that requires us-east-1 test account credentials (784379639078).
     import_model: mark a test that creates a Bedrock model import job. Concurrent model import jobs are capped at 1 by a non-raisable Bedrock service quota, so these run serially in a dedicated scheduled CI run, not in PR checks.
+    skip_in_pr_check: mark a test that is excluded from PR check runs. Long-running or hang-prone tests that would otherwise push the run past the CodeBuild timeout; they run in a dedicated scheduled CI run instead.
 
 [testenv]
 setenv =