From 8be45689d5d2de2201e1ca14a4c92eb54c81190a Mon Sep 17 00:00:00 2001 From: Aditi Sharma <165942273+Aditi2424@users.noreply.github.com> Date: Mon, 30 Dec 2024 09:37:35 -0800 Subject: [PATCH] fix: Disable jumpstart tests missing clean up logic (#4976) * Disable jumpstart tests missing clean up logic * Black format --------- Co-authored-by: adishaa Co-authored-by: parknate@ --- tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py | 2 ++ tests/integ/sagemaker/jumpstart/private_hub/test_hub.py | 1 + 2 files changed, 3 insertions(+) diff --git a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py index b938f489df..5e54c7551f 100644 --- a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py +++ b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py @@ -477,6 +477,7 @@ def _teardown_test_hub_with_reference(public_hub_model_id: str): raise e +@pytest.mark.skip # Currently JumpStartModel does not pull from HubService for the Public Hub. def test_model_reference_marketplace_model(setup): session = get_sm_session() @@ -535,6 +536,7 @@ def test_model_reference_marketplace_model(setup): # _teardown_test_hub_with_reference(public_hub_marketplace_model_id) +@pytest.mark.skip def test_bedrock_store_model_tags_from_hub_service(setup): session = get_sm_session() diff --git a/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py b/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py index 2bccb96524..db5d868c06 100644 --- a/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py +++ b/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py @@ -32,6 +32,7 @@ def hub_instance(): yield hub +@pytest.mark.skip def test_private_hub(setup, hub_instance): # Createhub create_hub_response = hub_instance.create(