fix: Add avg alias and fix aggregation group_by_keys

- Add "avg" -> "mean" function alias for aggregation
- Fix group_by_keys retrieval in LocalFeatureBuilder to use column_info.join_keys

Author: cutoutsy

sdk/python/feast/aggregation/__init__.py

@@ -120,6 +120,7 @@ def resolved_name(self, time_window: Optional[timedelta] = None) -> str:
 
 _FUNCTION_ALIASES: Dict[str, str] = {
     "count_distinct": "nunique",
+    "avg": "mean",
 }
 
 

sdk/python/feast/infra/compute_engines/local/feature_builder.py

@@ -65,7 +65,8 @@ def build_aggregation_node(self, view, input_node):
                 "Time window aggregation is not supported in the local compute engine."
             ),
         )
-        group_by_keys = view.entities
+        column_info = self.get_column_info(view)
+        group_by_keys = column_info.join_keys
         node = LocalAggregationNode(
             "agg", self.backend, group_by_keys, agg_ops, inputs=[input_node]
         )