From 6fc1faef7020edae133a97aad88f6dc63209727f Mon Sep 17 00:00:00 2001
From: Vinny Senthil <vinnysenthil@gmail.com>
Date: Thu, 22 Apr 2021 14:40:18 -0500
Subject: [PATCH] fix: Add all supported uCAIP GA regions

---
 google/cloud/aiplatform/constants.py      | 17 ++++++++++++++++-
 tests/unit/aiplatform/test_initializer.py |  5 +++++
 2 files changed, 21 insertions(+), 1 deletion(-)

diff --git a/google/cloud/aiplatform/constants.py b/google/cloud/aiplatform/constants.py
index 62c28009c2..9e71427d0f 100644
--- a/google/cloud/aiplatform/constants.py
+++ b/google/cloud/aiplatform/constants.py
@@ -16,7 +16,22 @@
 #
 
 DEFAULT_REGION = "us-central1"
-SUPPORTED_REGIONS = ("us-central1", "europe-west4", "asia-east1")
+SUPPORTED_REGIONS = {
+    "asia-east1",
+    "asia-northeast1",
+    "asia-northeast3",
+    "asia-southeast1",
+    "australia-southeast1",
+    "europe-west1",
+    "europe-west2",
+    "europe-west4",
+    "northamerica-northeast1",
+    "us-central1",
+    "us-east1",
+    "us-east4",
+    "us-west1",
+}
+
 API_BASE_PATH = "aiplatform.googleapis.com"
 
 # Batch Prediction
diff --git a/tests/unit/aiplatform/test_initializer.py b/tests/unit/aiplatform/test_initializer.py
index 1d97ad2e9a..91d86409f9 100644
--- a/tests/unit/aiplatform/test_initializer.py
+++ b/tests/unit/aiplatform/test_initializer.py
@@ -141,6 +141,11 @@ def test_create_client_user_agent(self):
             ("us-central1", None, "us-central1-aiplatform.googleapis.com"),
             ("us-central1", "europe-west4", "europe-west4-aiplatform.googleapis.com",),
             ("asia-east1", None, "asia-east1-aiplatform.googleapis.com"),
+            (
+                "asia-southeast1",
+                "australia-southeast1",
+                "australia-southeast1-aiplatform.googleapis.com",
+            ),
         ],
     )
     def test_get_client_options(