diff --git a/sdk/python/feast/infra/offline_stores/bigquery_source.py b/sdk/python/feast/infra/offline_stores/bigquery_source.py index 5b1291b4e17..536dbc5c141 100644 --- a/sdk/python/feast/infra/offline_stores/bigquery_source.py +++ b/sdk/python/feast/infra/offline_stores/bigquery_source.py @@ -162,6 +162,7 @@ def get_table_column_names_and_types( from google.api_core import client_info as http_client_info except ImportError as e: from feast.errors import FeastExtrasDependencyImportError + raise FeastExtrasDependencyImportError("gcp", str(e)) from google.cloud import bigquery diff --git a/setup.py b/setup.py index 4a2c0a47bd3..0d564ca265c 100644 --- a/setup.py +++ b/setup.py @@ -58,7 +58,7 @@ "numpy>=1.22,<3", "pandas>=1.4.3,<2", "pandavro~=1.5.0", # For some reason pandavro higher than 1.5.* only support pandas less than 1.3. - "protobuf<5,>3", + "protobuf<5,>3.20", "proto-plus>=1.20.0,<2", "pyarrow>=4,<9", "pydantic>=1,<2",