Skip to content

Commit e62e927

Browse files
committed
Merge remote-tracking branch 'upstream/master'
2 parents dd992d3 + bdf20bb commit e62e927

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

sdk/python/feast/utils.py

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -697,8 +697,13 @@ def _get_entity_maps(
697697
entity.join_key, entity.join_key
698698
)
699699
entity_name_to_join_key_map[entity_name] = join_key
700+
700701
for entity_column in feature_view.entity_columns:
701-
entity_type_map[entity_column.name] = entity_column.dtype.to_value_type()
702+
dtype = entity_column.dtype.to_value_type()
703+
entity_join_key_column_name = feature_view.projection.join_key_map.get(
704+
entity_column.name, entity_column.name
705+
)
706+
entity_type_map[entity_join_key_column_name] = dtype
702707

703708
return (
704709
entity_name_to_join_key_map,

0 commit comments

Comments
 (0)