Skip to content

Commit 8ca47bd

Browse files
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>
1 parent 36747aa commit 8ca47bd

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

sdk/python/feast/inference.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -199,10 +199,10 @@ def _infer_features_and_entities(
199199
fv.batch_source.timestamp_field,
200200
fv.batch_source.created_timestamp_column,
201201
}
202-
for column in columns_to_exclude:
203-
if column in fv.batch_source.field_mapping:
204-
columns_to_exclude.remove(column)
205-
columns_to_exclude.add(fv.batch_source.field_mapping[column])
202+
for original_col, mapped_col in fv.batch_source.field_mapping.items():
203+
if mapped_col in columns_to_exclude:
204+
columns_to_exclude.remove(mapped_col)
205+
columns_to_exclude.add(original_col)
206206

207207
table_column_names_and_types = fv.batch_source.get_table_column_names_and_types(
208208
config

0 commit comments

Comments
 (0)