diff --git a/google/cloud/aiplatform/constants/base.py b/google/cloud/aiplatform/constants/base.py index 5e09fce1d0..230918d564 100644 --- a/google/cloud/aiplatform/constants/base.py +++ b/google/cloud/aiplatform/constants/base.py @@ -36,6 +36,8 @@ "us-east4", "us-west1", "us-west2", + "us-west4", + "southamerica-east1", } API_BASE_PATH = "aiplatform.googleapis.com" diff --git a/tests/unit/aiplatform/test_utils.py b/tests/unit/aiplatform/test_utils.py index b47eb684d8..8308b04cca 100644 --- a/tests/unit/aiplatform/test_utils.py +++ b/tests/unit/aiplatform/test_utils.py @@ -42,7 +42,7 @@ def test_invalid_region_raises_with_invalid_region(): with pytest.raises(ValueError): - aiplatform.utils.validate_region(region="us-west4") + aiplatform.utils.validate_region(region="us-west3") def test_invalid_region_does_not_raise_with_valid_region():