diff --git a/sdk/python/feast/infra/offline_stores/snowflake.py b/sdk/python/feast/infra/offline_stores/snowflake.py index 5ab11cf3f2f..36815747d99 100644 --- a/sdk/python/feast/infra/offline_stores/snowflake.py +++ b/sdk/python/feast/infra/offline_stores/snowflake.py @@ -52,7 +52,7 @@ ) from feast.repo_config import FeastConfigBaseModel, RepoConfig from feast.saved_dataset import SavedDatasetStorage -from feast.types import Array, Float32, Float64, Int32, Int64, String, UnixTimestamp +from feast.types import Array, Float32, Float64, Int32, Int64, String, UnixTimestamp, Bool from feast.usage import log_exceptions_and_usage try: @@ -452,6 +452,7 @@ def _to_df_internal(self, timeout: Optional[int] = None) -> pd.DataFrame: Array(UnixTimestamp), Array(Float64), Array(Float32), + Array(Bool), ]: df[feature.name] = [json.loads(x) for x in df[feature.name]]