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 ff1c148e20e Small fix for feature_store system test + add output for
operator (#57605)
ff1c148e20e is described below
commit ff1c148e20e2cf62a32c51388d708592ce37e94c
Author: VladaZakharova <[email protected]>
AuthorDate: Mon Nov 3 00:38:55 2025 +0100
Small fix for feature_store system test + add output for operator (#57605)
---
.../providers/google/cloud/operators/vertex_ai/feature_store.py | 2 +-
.../google/cloud/vertex_ai/example_vertex_ai_feature_store.py | 7 +------
2 files changed, 2 insertions(+), 7 deletions(-)
diff --git
a/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/feature_store.py
b/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/feature_store.py
index f00a7ad1265..3f34963437e 100644
---
a/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/feature_store.py
+++
b/providers/google/src/airflow/providers/google/cloud/operators/vertex_ai/feature_store.py
@@ -179,7 +179,7 @@ class
CreateFeatureOnlineStoreOperator(GoogleCloudBaseOperator, OperationHelper)
gcp_conn_id=self.gcp_conn_id,
impersonation_chain=self.impersonation_chain,
)
- self.log.info("Creating the Feature Online Store...")
+ self.log.info("Creating the Feature Online Store id: %s",
self.feature_online_store_id)
result_operation = hook.create_feature_online_store(
project_id=self.project_id,
location=self.location,
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 1a88cddb6fd..c6806e6ed9d 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
@@ -23,8 +23,6 @@ Example Airflow DAG for Google Vertex AI Feature Store
operations.
from __future__ import annotations
import os
-import random
-import string
from datetime import datetime, timedelta
from google.cloud.aiplatform_v1beta1 import FeatureOnlineStore, FeatureView,
FeatureViewDataKey
@@ -64,10 +62,7 @@ BQ_VIEW_ID = "product_features_view"
BQ_VIEW_FQN = f"{PROJECT_ID}.{BQ_DATASET_ID}.{BQ_VIEW_ID}"
# Please take into consideration that max ID length is 60 symbols
-PREFIX_FEATURE_ONLINE_STORE_ID = "".join(
- random.choice(string.ascii_letters + string.digits) for _ in range(20)
-)
-FEATURE_ONLINE_STORE_ID =
f"feature_online_store_{PREFIX_FEATURE_ONLINE_STORE_ID}".replace("-", "_")
+FEATURE_ONLINE_STORE_ID = f"{ENV_ID}_fo_id".replace("-", "_")
FEATURE_VIEW_ID = "feature_view_product"
FEATURE_VIEW_DATA_KEY = {"key": "28098"}