Skip to content
Merged
Prev Previous commit
Next Next commit
prune out unneeded fields in push source
Signed-off-by: Danny Chiao <danny@tecton.ai>
  • Loading branch information
adchia committed Apr 18, 2022
commit 315f2d84259326e34be4d4e57f1f79fbe4e3c3ce
7 changes: 1 addition & 6 deletions sdk/python/feast/feature_store.py
Original file line number Diff line number Diff line change
Expand Up @@ -1209,13 +1209,8 @@ def push(
}

for fv in fvs_with_push_sources:
push_source = cast(PushSource, fv.stream_source)
# If the push source's timestamp field doesn't match the batch source's timestamp field, do the mapping
renamed_df = df
if len(push_source.timestamp_field) > 0:
renamed_df = df.rename(columns={push_source.timestamp_field: push_source.batch_source.timestamp_field})
self.write_to_online_store(
fv.name, renamed_df, allow_registry_cache=allow_registry_cache
fv.name, df, allow_registry_cache=allow_registry_cache
)

@log_exceptions_and_usage
Expand Down