refactor: Don't enter streaming engine for groupby-> agg mean/median …

…date
pola-rs · Jun 7, 2024 · 6a48e65 · 6a48e65
1 parent faf0f06
commit 6a48e65
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/crates/polars-pipe/src/executors/sinks/group_by/aggregates/convert.rs b/crates/polars-pipe/src/executors/sinks/group_by/aggregates/convert.rs
@@ -101,7 +101,12 @@ pub fn can_convert_to_hash_agg(
                     }
                 ) && {
                     if let Ok(field) = ae.to_field(input_schema, Context::Default, expr_arena) {
-                        field.dtype.to_physical().is_numeric()
+                        match field.dtype {
+                            DataType::Date => {
+                                matches!(agg_fn, IRAggExpr::Mean(_) | IRAggExpr::Median(_))
+                            },
+                            _ => field.dtype.to_physical().is_numeric(),
+                        }
                     } else {
                         false
                     }