fix: Add all supported uCAIP GA regions (#350)

googleapis · Apr 22, 2021 · 5e14c59 · 5e14c59
1 parent 6f6c2dd
commit 5e14c59
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git 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
@@ -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(