feat: Adding spot, reservation_affinity to Vertex SDK

PiperOrigin-RevId: 660930164
googleapis · Aug 8, 2024 · 3e785bd · 3e785bd
1 parent d6ec6c2
commit 3e785bd
Show file tree

Hide file tree

Showing 4 changed files with 413 additions and 4 deletions.
diff --git a/google/cloud/aiplatform/compat/types/__init__.py b/google/cloud/aiplatform/compat/types/__init__.py
@@ -93,6 +93,7 @@
     pipeline_state as pipeline_state_v1beta1,
     prediction_service as prediction_service_v1beta1,
     publisher_model as publisher_model_v1beta1,
+    reservation_affinity as reservation_affinity_v1beta1,
     service_networking as service_networking_v1beta1,
     schedule as schedule_v1beta1,
     schedule_service as schedule_service_v1beta1,
@@ -176,6 +177,7 @@
     pipeline_state as pipeline_state_v1,
     prediction_service as prediction_service_v1,
     publisher_model as publisher_model_v1,
+    reservation_affinity as reservation_affinity_v1,
     schedule as schedule_v1,
     schedule_service as schedule_service_v1,
     service_networking as service_networking_v1,
@@ -254,6 +256,7 @@
     pipeline_state_v1,
     prediction_service_v1,
     publisher_model_v1,
+    reservation_affinity_v1,
     schedule_v1,
     schedule_service_v1,
     specialist_pool_v1,
@@ -337,6 +340,7 @@
     pipeline_state_v1beta1,
     prediction_service_v1beta1,
     publisher_model_v1beta1,
+    reservation_affinity_v1beta1,
     schedule_v1beta1,
     schedule_service_v1beta1,
     specialist_pool_v1beta1,