fix test

Signed-off-by: HaoXuAI <sduxuhao@gmail.com>
feast-dev · franciscojavierarceo · Jun 3, 2025 · May 14, 2025 · May 14, 2025 · May 15, 2025
commit 96e99b681ea12f74910e470bbc7439fcc3d0ff58
@@ -156,7 +156,7 @@ def execute(self, context: ExecutionContext) -> ArrowTableValue:
         dedup_keys = context.column_info.join_keys
         if dedup_keys:
             sort_keys = [column_info.timestamp_column]
-            if column_info.created_timestamp_column:
+            if column_info.created_timestamp_column and column_info.created_timestamp_column in df.columns:
                 sort_keys.append(column_info.created_timestamp_column)
 
             df = self.backend.drop_duplicates(

diff --git a/sdk/python/feast/infra/offline_stores/ibis.py b/sdk/python/feast/infra/offline_stores/ibis.py
@@ -296,6 +296,12 @@ def pull_all_from_table_or_query_ibis(
         )
     )
 
+    if data_source.field_mapping:
+        for old_field_name in data_source.field_mapping:
+            table = table.rename(
+                {data_source.field_mapping[old_field_name]: old_field_name}
+            )
+
     return IbisRetrievalJob(
         table=table,
         on_demand_feature_views=[],

@@ -358,6 +358,7 @@ def pull_all_from_table_or_query(self, command: dict):
             data_source=data_source,
             join_key_columns=command["join_key_columns"],
             feature_name_columns=command["feature_name_columns"],
+            created_timestamp_column=command["created_timestamp_column"],
             timestamp_field=command["timestamp_field"],
             start_date=utils.make_tzaware(
                 datetime.fromisoformat(command["start_date"])