From 7e1763eb1eb35c73a319801c0cf2d150dd815592 Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Fri, 19 Dec 2025 10:27:10 -0800
Subject: [PATCH 1/2] Skipping expensive integ tests

---
 .../inference/cli/test_cli_jumpstart_inference_with_mig.py       | 1 +
 .../inference/sdk/test_sdk_jumpstart_inference_with_mig.py       | 1 +
 2 files changed, 2 insertions(+)

diff --git a/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py b/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
index bd385290..782c3bf9 100644
--- a/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
+++ b/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
@@ -28,6 +28,7 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- JumpStart Endpoint Tests ---------
+@pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
 @pytest.mark.dependency(name="create")
 def test_js_create(runner, js_endpoint_name):
     result = runner.invoke(js_create, [
diff --git a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
index f8645352..34e70aea 100644
--- a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
+++ b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
@@ -38,6 +38,7 @@ def endpoint_obj():
 
     return HPJumpStartEndpoint(metadata=metadata, model=model, server=server, sage_maker_endpoint=sm_endpoint)
 
+@pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
 @pytest.mark.dependency(name="create")
 def test_create_endpoint(endpoint_obj):
     endpoint_obj.create()

From b13dc248611cf01929f978f193fbc3ab9a74b616 Mon Sep 17 00:00:00 2001
From: aviruthen <91846056+aviruthen@users.noreply.github.com>
Date: Fri, 19 Dec 2025 10:36:10 -0800
Subject: [PATCH 2/2] Properly skipping all tests

---
 .../inference/cli/test_cli_jumpstart_inference_with_mig.py     | 3 ++-
 .../inference/sdk/test_sdk_jumpstart_inference_with_mig.py     | 3 ++-
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py b/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
index 782c3bf9..8bb06f00 100644
--- a/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
+++ b/test/integration_tests/inference/cli/test_cli_jumpstart_inference_with_mig.py
@@ -8,6 +8,8 @@
 from sagemaker.hyperpod.inference.hp_jumpstart_endpoint import HPJumpStartEndpoint
 from test.integration_tests.utils import get_time_str
 
+pytestmark = pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
+
 # --------- Test Configuration ---------
 NAMESPACE = "integration"
 VERSION = "1.1"
@@ -28,7 +30,6 @@ def sagemaker_client():
     return boto3.client("sagemaker", region_name=REGION)
 
 # --------- JumpStart Endpoint Tests ---------
-@pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
 @pytest.mark.dependency(name="create")
 def test_js_create(runner, js_endpoint_name):
     result = runner.invoke(js_create, [
diff --git a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
index 34e70aea..ae178256 100644
--- a/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
+++ b/test/integration_tests/inference/sdk/test_sdk_jumpstart_inference_with_mig.py
@@ -9,6 +9,8 @@
 from test.integration_tests.utils import get_time_str
 from sagemaker.hyperpod.common.config.metadata import Metadata
 
+pytestmark = pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
+
 # --------- Config ---------
 NAMESPACE = "integration"
 REGION = "us-east-2"
@@ -38,7 +40,6 @@ def endpoint_obj():
 
     return HPJumpStartEndpoint(metadata=metadata, model=model, server=server, sage_maker_endpoint=sm_endpoint)
 
-@pytest.mark.skip(reason="Skipping expensive ml.p4d.24xlarge MIG tests to reduce costs")
 @pytest.mark.dependency(name="create")
 def test_create_endpoint(endpoint_obj):
     endpoint_obj.create()