Merge pull request #26 from cleardataeng/feature/add-readiness-check-…

…vrtx-ai Add readiness and terminal values for vertex ai instances.
cleardataeng · Dec 19, 2024 · ce944b5 · ce944b5
2 parents 6f625ad + 02a3086
commit ce944b5
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml
@@ -12,7 +12,7 @@ jobs:
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        python-version: [3.7, 3.8, 3.9]
+        python-version: [3.8, 3.9]
 
     steps:
     - uses: actions/checkout@v2

diff --git a/rpe/resources/gcp.py b/rpe/resources/gcp.py
@@ -1157,6 +1157,17 @@ class GcpNotebookInstance(GoogleAPIResource):
     resource_path = "projects.locations.instances"
     version = "v2"
 
+    readiness_key = "state"
+    readiness_value = "ACTIVE"
+    readiness_terminal_values = [
+        "STATE_UNSPECIFIED",
+        "STOPPING",
+        "STOPPED",
+        "DELETED",
+        "SUSPENDING",
+        "SUSPENDED",
+    ]
+
     required_resource_data = ["name", "location", "project_id"]
 
     resource_type = "notebooks.googleapis.com/Instance"