We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 36747aa commit 8ca47bdCopy full SHA for 8ca47bd
sdk/python/feast/inference.py
@@ -199,10 +199,10 @@ def _infer_features_and_entities(
199
fv.batch_source.timestamp_field,
200
fv.batch_source.created_timestamp_column,
201
}
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])
+ 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)
206
207
table_column_names_and_types = fv.batch_source.get_table_column_names_and_types(
208
config
0 commit comments