From ca696fbeac1f7be841bcc53f1f8aac3ec042979c Mon Sep 17 00:00:00 2001 From: mrishi Date: Tue, 12 May 2026 12:22:48 +0530 Subject: [PATCH] fix(feature_store): safely parse bq:// URIs using removeprefix The previous implementation used `.lstrip("bq://")`, which greedily dropped valid characters (like 'b' or 'q') from the beginning of Google Cloud Project IDs. Replaced `lstrip` with `removeprefix` to explicitly remove only the protocol string without mutating the project ID. --- vertexai/resources/preview/feature_store/offline_store.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/vertexai/resources/preview/feature_store/offline_store.py b/vertexai/resources/preview/feature_store/offline_store.py index 3a785711e4..ad5bf193e7 100644 --- a/vertexai/resources/preview/feature_store/offline_store.py +++ b/vertexai/resources/preview/feature_store/offline_store.py @@ -118,7 +118,8 @@ def _feature_to_data_source( bq_uri = feature_group._gca_resource.big_query.big_query_source.input_uri assert bq_uri - fully_qualified_table = bq_uri.lstrip("bq://") + fully_qualified_table = bq_uri.removeprefix("bq://") + assert fully_qualified_table query = (