diff --git a/agentplatform/resources/preview/feature_store/offline_store.py b/agentplatform/resources/preview/feature_store/offline_store.py index 4ba369d168..1e55ae127c 100644 --- a/agentplatform/resources/preview/feature_store/offline_store.py +++ b/agentplatform/resources/preview/feature_store/offline_store.py @@ -116,7 +116,7 @@ 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 = ( diff --git a/google/cloud/aiplatform/preview/datasets.py b/google/cloud/aiplatform/preview/datasets.py index 1616abb991..5e0bde0340 100644 --- a/google/cloud/aiplatform/preview/datasets.py +++ b/google/cloud/aiplatform/preview/datasets.py @@ -1409,7 +1409,7 @@ def assemble( ) result = assemble_lro.result(timeout=None) _LOGGER.log_action_completed_against_resource("data", "assembled", self) - table_id = result.bigquery_destination.lstrip("bq://") + table_id = result.bigquery_destination.removeprefix("bq://") if load_dataframe: session_options = bigframes.BigQueryOptions( credentials=initializer.global_config.credentials, 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 = (