Fix bug in event timestamp removal in local mode (#1441)

Signed-off-by: Jacob Klegar <jacob@tecton.ai>
feast-dev · Apr 8, 2021 · 2815c66 · 2815c66
1 parent f20ccfa
commit 2815c66
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/sdk/python/feast/infra/local_sqlite.py b/sdk/python/feast/infra/local_sqlite.py
@@ -279,9 +279,10 @@ def evaluate_historical_retrieval():
                 )
 
                 # Remove right (feature table/view) event_timestamp column.
-                entity_df_with_features.drop(
-                    columns=[event_timestamp_column], inplace=True
-                )
+                if event_timestamp_column != ENTITY_DF_EVENT_TIMESTAMP_COL:
+                    entity_df_with_features.drop(
+                        columns=[event_timestamp_column], inplace=True
+                    )
 
                 # Ensure that we delete dataframes to free up memory
                 del df_to_join