Include List Bool in pandas transformation

Signed-off-by: john.lemmon <john.lemmon@medely.com>
feast-dev · Sep 18, 2023 · 9f5a124 · 9f5a124
1 parent f01892a
commit 9f5a124
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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]]