From a89532ce206e18921ad70c74d544fa2e7af9c94f Mon Sep 17 00:00:00 2001 From: Gbenga Olayiwola Date: Thu, 6 Aug 2026 13:32:53 +0000 Subject: [PATCH] Fix Vertex AI Feature Store system test failure Configure Bigtable autoscaling for the Feature Online Store and add TriggerRule.ALL_DONE to teardown tasks to prevent leaked resources on test failure. --- .../vertex_ai/example_vertex_ai_feature_store.py | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/providers/google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py b/providers/google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py index 2d88c8adbb814..52e8b649064a1 100644 --- a/providers/google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py +++ b/providers/google/tests/system/google/cloud/vertex_ai/example_vertex_ai_feature_store.py @@ -127,7 +127,15 @@ project_id=PROJECT_ID, location=REGION, feature_online_store_id=FEATURE_ONLINE_STORE_ID, - feature_online_store=FeatureOnlineStore(optimized=FeatureOnlineStore.Optimized()), + feature_online_store=FeatureOnlineStore( + bigtable=FeatureOnlineStore.Bigtable( + auto_scaling=FeatureOnlineStore.Bigtable.AutoScaling( + min_node_count=1, + max_node_count=3, + cpu_utilization_target=50, + ) + ) + ), ) # [END how_to_cloud_vertex_ai_create_feature_online_store_operator] @@ -206,6 +214,7 @@ location=REGION, feature_online_store_id=FEATURE_ONLINE_STORE_ID, feature_view_id=FEATURE_VIEW_ID, + trigger_rule=TriggerRule.ALL_DONE, ) # [END how_to_cloud_vertex_ai_delete_feature_view_operator] @@ -215,6 +224,7 @@ project_id=PROJECT_ID, location=REGION, feature_online_store_id=FEATURE_ONLINE_STORE_ID, + trigger_rule=TriggerRule.ALL_DONE, ) # [END how_to_cloud_vertex_ai_delete_feature_online_store_operator]