From df668e7f50191dcf5be068dbbadcd28c3239b5f0 Mon Sep 17 00:00:00 2001
From: Weijie Guo <reswqa@163.com>
Date: Fri, 12 Apr 2024 16:46:46 +0800
Subject: [PATCH] fix

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

diff --git a/crates/polars-lazy/src/physical_plan/expressions/apply.rs b/crates/polars-lazy/src/physical_plan/expressions/apply.rs
index 405bc92c0654..1bfb9f12a035 100644
--- a/crates/polars-lazy/src/physical_plan/expressions/apply.rs
+++ b/crates/polars-lazy/src/physical_plan/expressions/apply.rs
@@ -365,10 +365,8 @@ impl PhysicalExpr for ApplyExpr {
                             _ => {},
                         }
                     }
-                    if has_agg_list {
+                    if has_agg_list || (has_agg_scalar && has_not_agg) {
                         return self.apply_multiple_group_aware(acs, df);
-                    } else if has_agg_scalar && has_not_agg {
-                        self.apply_multiple_group_aware(acs, df)
                     } else {
                         apply_multiple_elementwise(
                             acs,