add init_auto

googleapis · Apr 26, 2024 · 3c3c5b4 · 3c3c5b4
1 parent e0f5601
commit 3c3c5b4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/google/cloud/aiplatform/initializer.py b/google/cloud/aiplatform/initializer.py
@@ -109,6 +109,11 @@ def __init__(self):
         self._api_endpoint = None
         self._api_transport = None
 
+    def init_auto(self, *):
+        project = os.environ.get("GOOGLE_CLOUD_PROJECT")
+        location = os.getenv("GOOGLE_CLOUD_REGION") or os.getenv("CLOUD_ML_REGION")
+        self.init(project=project, location=location)
+
     def init(
         self,
         *,