Skip to content

Commit c350dd0

Browse files
leiyangyouHaoXuAI
authored andcommitted
fix: Correct entity value type mapping for aliased feature views (#5492)
1 parent 048f34e commit c350dd0

1 file changed

Lines changed: 6 additions & 1 deletion

File tree

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)