test(serve): Add skip_in_pr_check marker for hang-prone integ tests

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

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

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.
 

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

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 =