diff --git a/tests/integ/__init__.py b/tests/integ/__init__.py index 120311aab9..90f4029a92 100644 --- a/tests/integ/__init__.py +++ b/tests/integ/__init__.py @@ -113,9 +113,6 @@ "eu-south-1", ] NO_AUTO_ML_REGIONS = [ - "sa-east-1", - "me-south-1", - "ap-east-1", "eu-west-3", "af-south-1", "eu-south-1", diff --git a/tests/integ/test_auto_ml.py b/tests/integ/test_auto_ml.py index fc547256c6..79054c2d4f 100644 --- a/tests/integ/test_auto_ml.py +++ b/tests/integ/test_auto_ml.py @@ -18,7 +18,7 @@ import tests.integ from sagemaker import AutoML, CandidateEstimator, AutoMLInput -from sagemaker.exceptions import UnexpectedStatusException +from botocore.exceptions import ClientError from sagemaker.utils import unique_name_from_base from tests.integ import DATA_DIR, AUTO_ML_DEFAULT_TIMEMOUT_MINUTES, auto_ml_utils from tests.integ.timeout import timeout @@ -142,7 +142,9 @@ def test_auto_ml_invalid_target_attribute(sagemaker_session): job_name = unique_name_from_base("auto-ml", max_length=32) inputs = sagemaker_session.upload_data(path=TRAINING_DATA, key_prefix=PREFIX + "/input") with pytest.raises( - UnexpectedStatusException, match="Could not complete the data builder processing job." + ClientError, + match=r"An error occurred \(ValidationException\) when calling the CreateAutoMLJob " + "operation: Target attribute name y does not exist in header.", ): auto_ml.fit(inputs, job_name=job_name)