From b08e03a173bec8a2a7e94ff4d5df4f77e666299f Mon Sep 17 00:00:00 2001 From: adishaa Date: Mon, 23 Dec 2024 12:13:49 -0800 Subject: [PATCH 1/3] 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/3] 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 From 7657dcd4d276fdcecab6b98249ff7be316d313dd Mon Sep 17 00:00:00 2001 From: adishaa Date: Thu, 1 May 2025 10:20:40 -0700 Subject: [PATCH 3/3] Fix integ tests --- tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py b/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py index ea65f998c8..92199e0efd 100644 --- a/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py +++ b/tests/integ/sagemaker/serve/test_serve_js_deep_unit_tests.py @@ -64,6 +64,7 @@ def test_js_model_with_optimize_speculative_decoding_config_gated_requests_are_e "Image": ANY, "Environment": { "SAGEMAKER_PROGRAM": "inference.py", + "SAGEMAKER_MODEL_SERVER_TIMEOUT": "3600", "ENDPOINT_SERVER_TIMEOUT": "3600", "MODEL_CACHE_ROOT": "/opt/ml/model", "SAGEMAKER_ENV": "1", @@ -150,6 +151,7 @@ def test_js_model_with_optimize_sharding_and_resource_requirements_requests_are_ "Image": ANY, "Environment": { "SAGEMAKER_PROGRAM": "inference.py", + "SAGEMAKER_MODEL_SERVER_TIMEOUT": "3600", "ENDPOINT_SERVER_TIMEOUT": "3600", "MODEL_CACHE_ROOT": "/opt/ml/model", "SAGEMAKER_ENV": "1",