c

pola-rs · Nov 5, 2024 · 6c883d0 · 6c883d0
1 parent ae513be
commit 6c883d0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/crates/polars-expr/src/expressions/apply.rs b/crates/polars-expr/src/expressions/apply.rs
@@ -170,7 +170,7 @@ impl ApplyExpr {
                 // })?
                 let out: ListChunked = POOL.install(|| iter.collect::<PolarsResult<_>>())?;
 
-                debug_assert_eq!(out.dtype(), &dtype);
+                debug_assert_eq!(&dtype, out.dtype());
 
                 out
             } else {

diff --git a/crates/polars-expr/src/planner.rs b/crates/polars-expr/src/planner.rs
@@ -571,9 +571,15 @@ fn create_physical_expr_inner(
                 move |c: &mut [polars_core::frame::column::Column]| c[0].explode().map(Some),
             ) as Arc<dyn ColumnsUdf>);
 
-            let field = expr_arena
-                .get(expression)
-                .to_field(schema, ctxt, expr_arena)?;
+            let mut field =
+                expr_arena
+                    .get(expression)
+                    .to_field(schema, Context::Default, expr_arena)?;
+
+            if let Context::Aggregation = ctxt {
+                field.dtype = DataType::List(Box::new(field.dtype));
+            }
+
             Ok(Arc::new(ApplyExpr::new(
                 vec![input],
                 function,