Skip to content

Commit 934c1e9

Browse files
okay think it is working
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
1 parent b92bf32 commit 934c1e9

1 file changed

Lines changed: 10 additions & 1 deletion

File tree

sdk/python/feast/infra/passthrough_provider.py

Lines changed: 10 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -294,6 +294,14 @@ def ingest_df(
294294
self.repo_config, feature_view, rows_to_write, progress=None
295295
)
296296
else:
297+
if hasattr(feature_view, 'entity_columns'):
298+
join_keys = {
299+
entity.name: entity.dtype.to_value_type()
300+
for entity in feature_view.entity_columns
301+
}
302+
else:
303+
join_keys = {}
304+
297305
# Note: A dictionary mapping of column names in this data
298306
# source to feature names in a feature table or view. Only used for feature
299307
# columns, not entity or timestamp columns.
@@ -302,8 +310,9 @@ def ingest_df(
302310
table = _run_pyarrow_field_mapping(
303311
table, feature_view.batch_source.field_mapping
304312
)
313+
else:
314+
table = _run_pyarrow_field_mapping(table, {})
305315

306-
join_keys = {}
307316
if not isinstance(feature_view, BaseFeatureView):
308317
for entity in feature_view.entity_columns:
309318
join_keys[entity.name] = entity.dtype.to_value_type()

0 commit comments

Comments
 (0)