Skip to content

Commit

Permalink
fix: Fix field mapping logic during feature inference (#3067)
Browse files Browse the repository at this point in the history
fix: Fix bug where feature inference was improperly registering timestamp and created field if part of field_mapping

Signed-off-by: Miles Adkins <miles.adkins@snowflake.com>

Signed-off-by: Miles Adkins <miles.adkins@snowflake.com>
  • Loading branch information
sfc-gh-madkins authored and adchia committed Aug 15, 2022
1 parent bdc3451 commit 3668702
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions sdk/python/feast/inference.py
Expand Up @@ -192,10 +192,10 @@ def _infer_features_and_entities(
fv.batch_source.timestamp_field,
fv.batch_source.created_timestamp_column,
}
for column in columns_to_exclude:
if column in fv.batch_source.field_mapping:
columns_to_exclude.remove(column)
columns_to_exclude.add(fv.batch_source.field_mapping[column])
for original_col, mapped_col in fv.batch_source.field_mapping.items():
if mapped_col in columns_to_exclude:
columns_to_exclude.remove(mapped_col)
columns_to_exclude.add(original_col)

table_column_names_and_types = fv.batch_source.get_table_column_names_and_types(
config
Expand Down

0 comments on commit 3668702

Please sign in to comment.