This is an automated email from the ASF dual-hosted git repository.
potiuk pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new 015313a2328 Fix Vertex AI Feature Store system test failure (#71238)
015313a2328 is described below
commit 015313a232801228dcb91dba9f504cd932dc2185
Author: Gbenga <[email protected]>
AuthorDate: Wed Aug 19 11:30:14 2026 +0200
Fix Vertex AI Feature Store system test failure (#71238)
Configure Bigtable autoscaling for the Feature Online Store and add
TriggerRule.ALL_DONE to teardown tasks to prevent leaked resources on test
failure.
Co-authored-by: Gbenga Olayiwola <[email protected]>
---
.../cloud/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 2d88c8adbb8..52e8b649064 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 @@ with DAG(
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 @@ with DAG(
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 @@ with DAG(
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]
