From 6dd5e85b3d09a0cb7b7c273f6d8f675df4e428b5 Mon Sep 17 00:00:00 2001
From: lutengda <18346072982@163.com>
Date: Thu, 15 Aug 2024 11:01:15 +0800
Subject: [PATCH] when the limit pushdown is complete, remove the limit logical
 plan

---
 datafusion/optimizer/src/push_down_limit.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/datafusion/optimizer/src/push_down_limit.rs b/datafusion/optimizer/src/push_down_limit.rs
index 300ecdc4fb99..ad5355bd5ec1 100644
--- a/datafusion/optimizer/src/push_down_limit.rs
+++ b/datafusion/optimizer/src/push_down_limit.rs
@@ -127,7 +127,7 @@ impl OptimizerRule for PushDownLimit {
                         projected_schema: scan.projected_schema.clone(),
                         agg_with_grouping: scan.agg_with_grouping.clone(),
                     });
-                    Some(plan.with_new_inputs(&[new_input])?)
+                    Some(new_input)
                 }
             }
             LogicalPlan::Union(union) => {