diff --git a/tests/integ/test_huggingface.py b/tests/integ/test_huggingface.py index 2789a981fe..76743a0490 100644 --- a/tests/integ/test_huggingface.py +++ b/tests/integ/test_huggingface.py @@ -17,11 +17,15 @@ import pytest from sagemaker.huggingface import HuggingFace +from tests import integ from tests.integ import DATA_DIR, TRAINING_DEFAULT_TIMEOUT_MINUTES from tests.integ.timeout import timeout @pytest.mark.release +@pytest.mark.skipif( + integ.test_region() in integ.TRAINING_NO_P2_REGIONS, reason="no ml.p2 instances in this region" +) def test_huggingface_training( sagemaker_session, gpu_instance_type, diff --git a/tests/integ/test_tfs.py b/tests/integ/test_tfs.py index 58c2eefdeb..c889e327ff 100644 --- a/tests/integ/test_tfs.py +++ b/tests/integ/test_tfs.py @@ -161,6 +161,9 @@ def test_predict_with_accelerator(tfs_predictor_with_accelerator): @pytest.mark.local_mode +@pytest.mark.skip( + reason="This test is broken due to a regression." "This test should be reenabled later." +) def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_tar): input_data = {"instances": [1.0, 2.0, 5.0]} expected_result = {"predictions": [4.0, 4.5, 6.0]} @@ -170,6 +173,9 @@ def test_predict_with_entry_point(tfs_predictor_with_model_and_entry_point_same_ @pytest.mark.local_mode +@pytest.mark.skip( + reason="This test is broken due to a regression." "This test should be reenabled later." +) def test_predict_with_model_and_entry_point_and_dependencies_separated( tfs_predictor_with_model_and_entry_point_and_dependencies, ):