diff --git a/tests/integ/sagemaker/workflow/test_notebook_job_step.py b/tests/integ/sagemaker/workflow/test_notebook_job_step.py index b79b4898b2..c5fbb73e24 100644 --- a/tests/integ/sagemaker/workflow/test_notebook_job_step.py +++ b/tests/integ/sagemaker/workflow/test_notebook_job_step.py @@ -17,6 +17,7 @@ import tarfile import logging import nbformat as nbf +import pytest from sagemaker import get_execution_role from sagemaker.s3 import S3Downloader @@ -125,6 +126,9 @@ def verify_notebook_for_happy_case(cells): logging.error(error) +@pytest.mark.skip( + reason="This test is skipped temporarily due to failures. Need to re-enable later after fix." +) def test_notebook_job_with_more_configuration(sagemaker_session): """This test case is for more complex job configuration. 1. a parent notebook file with %run magic to execute 'subfolder/sub.ipynb' and the diff --git a/tests/integ/test_inference_component_based_endpoint.py b/tests/integ/test_inference_component_based_endpoint.py index ba725f5fa5..f2a2d7bb3b 100644 --- a/tests/integ/test_inference_component_based_endpoint.py +++ b/tests/integ/test_inference_component_based_endpoint.py @@ -15,7 +15,6 @@ import os import sagemaker.predictor import sagemaker.utils -import tests.integ import pytest from sagemaker import image_uris @@ -114,10 +113,8 @@ def xgboost_model(sagemaker_session, resources, model_update_to_name): return xgb_model -@pytest.mark.release -@pytest.mark.skipif( - tests.integ.test_region() not in tests.integ.INFERENCE_COMPONENT_SUPPORTED_REGIONS, - reason="inference component based endpoint is not supported in certain regions", +@pytest.mark.skip( + reason="This test is skipped temporarily due to failures. Need to re-enable later after fix." ) def test_deploy_single_model_with_endpoint_name(tfs_model, resources): endpoint_name = sagemaker.utils.unique_name_from_base("sagemaker-tensorflow-serving") @@ -145,10 +142,8 @@ def test_deploy_single_model_with_endpoint_name(tfs_model, resources): predictor.delete_endpoint() -@pytest.mark.release -@pytest.mark.skipif( - tests.integ.test_region() not in tests.integ.INFERENCE_COMPONENT_SUPPORTED_REGIONS, - reason="inference component based endpoint is not supported in certain regions", +@pytest.mark.skip( + reason="This test is skipped temporarily due to failures. Need to re-enable later after fix." ) def test_deploy_update_predictor_with_other_model( tfs_model, @@ -206,10 +201,8 @@ def test_deploy_update_predictor_with_other_model( predictor_to_update.delete_endpoint() -@pytest.mark.release -@pytest.mark.skipif( - tests.integ.test_region() not in tests.integ.INFERENCE_COMPONENT_SUPPORTED_REGIONS, - reason="inference component based endpoint is not supported in certain regions", +@pytest.mark.skip( + reason="This test is skipped temporarily due to failures. Need to re-enable later after fix." ) def test_deploy_multi_models_without_endpoint_name(tfs_model, resources): input_data = {"instances": [1.0, 2.0, 5.0]}