From d59c9301c486182faf015693966a34f207056504 Mon Sep 17 00:00:00 2001
From: Weijie Guo <reswqa@163.com>
Date: Fri, 12 Apr 2024 14:16:36 +0800
Subject: [PATCH] propagate scalar

---
 crates/polars-lazy/src/physical_plan/expressions/apply.rs | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/crates/polars-lazy/src/physical_plan/expressions/apply.rs b/crates/polars-lazy/src/physical_plan/expressions/apply.rs
index cb6e15e43a50..3b097e54d9c6 100644
--- 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(),
             ));