Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 0 additions & 3 deletions tests/integ/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
6 changes: 4 additions & 2 deletions tests/integ/test_auto_ml.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down Expand Up @@ -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)

Expand Down