fix: transformations

quintoandar · Oct 8, 2024 · cafe64d · cafe64d
1 parent 8c865c6
commit cafe64d
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/butterfree/transform/aggregated_feature_set.py b/butterfree/transform/aggregated_feature_set.py
@@ -576,11 +576,10 @@ def construct(
 
         pre_hook_df = self.run_pre_hooks(dataframe)
 
-        # Apply transformations
+        output_df = pre_hook_df
         for feature in self.keys + [self.timestamp]:
-            output_df = feature.transform(pre_hook_df)
+            output_df = feature.transform(output_df)
 
-        # Early filter data
         output_df = self.incremental_strategy.filter_with_incremental_strategy(
             dataframe=output_df, start_date=start_date, end_date=end_date
         )