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 2fe98b75221d4..defa1ff4e21e3 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 @@ def execute(self, context: Context) -> dict[str, Any]: 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 1a88cddb6fd06..c6806e6ed9d92 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 @@ 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_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"}