propagate scalar

pola-rs · Apr 12, 2024 · d59c930 · d59c930
1 parent 4085dd5
commit d59c930
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/crates/polars-lazy/src/physical_plan/expressions/apply.rs b/crates/polars-lazy/src/physical_plan/expressions/apply.rs
@@ -96,7 +96,9 @@ impl ApplyExpr {
         ca: ListChunked,
     ) -> PolarsResult<AggregationContext<'a>> {
         let all_unit_len = all_unit_length(&ca);
-        if all_unit_len && self.returns_scalar {
+        if all_unit_len
+            && (self.returns_scalar || matches!(ac.state, AggState::AggregatedScalar(_)))
+        {
             ac.with_agg_state(AggState::AggregatedScalar(
                 ca.explode().unwrap().into_series(),
             ));