From b08e03a173bec8a2a7e94ff4d5df4f77e666299f Mon Sep 17 00:00:00 2001 From: adishaa Date: Mon, 23 Dec 2024 12:13:49 -0800 Subject: [PATCH 1/2] Disable jumpstart tests missing clean up logic --- tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py | 4 ++-- tests/integ/sagemaker/jumpstart/private_hub/test_hub.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py index b938f489df..18dee74983 100644 --- a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py +++ b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py @@ -476,7 +476,7 @@ def _teardown_test_hub_with_reference(public_hub_model_id: str): else: 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() @@ -534,7 +534,7 @@ def test_model_reference_marketplace_model(setup): # predictor.delete_predictor() # _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..17110fd120 100644 --- a/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py +++ b/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py @@ -31,7 +31,7 @@ def hub_instance(): hub = Hub(HUB_NAME, sagemaker_session=get_sm_session()) yield hub - +@pytest.mark.skip def test_private_hub(setup, hub_instance): # Createhub create_hub_response = hub_instance.create( From acdbdba8f46f96647eb7125895bde4fbaabad34f Mon Sep 17 00:00:00 2001 From: adishaa Date: Mon, 23 Dec 2024 14:37:05 -0800 Subject: [PATCH 2/2] Black format --- 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 18dee74983..5e54c7551f 100644 --- a/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py +++ b/tests/integ/sagemaker/jumpstart/model/test_jumpstart_model.py @@ -476,6 +476,7 @@ def _teardown_test_hub_with_reference(public_hub_model_id: str): else: raise e + @pytest.mark.skip # Currently JumpStartModel does not pull from HubService for the Public Hub. def test_model_reference_marketplace_model(setup): @@ -534,6 +535,7 @@ def test_model_reference_marketplace_model(setup): # predictor.delete_predictor() # _teardown_test_hub_with_reference(public_hub_marketplace_model_id) + @pytest.mark.skip def test_bedrock_store_model_tags_from_hub_service(setup): diff --git a/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py b/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py index 17110fd120..db5d868c06 100644 --- a/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py +++ b/tests/integ/sagemaker/jumpstart/private_hub/test_hub.py @@ -31,6 +31,7 @@ def hub_instance(): hub = Hub(HUB_NAME, sagemaker_session=get_sm_session()) yield hub + @pytest.mark.skip def test_private_hub(setup, hub_instance): # Createhub