From f1c2dbe3b17a0d5edffbb65b85b642d0bb2756c5 Mon Sep 17 00:00:00 2001
From: Daniel Baumann <daniel.baumann@progress-linux.org>
Date: Tue, 19 Dec 2023 12:01:55 +0100
Subject: Merging upstream version 20.3.0.

Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
---
 sqlglot/executor/python.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'sqlglot/executor')

diff --git a/sqlglot/executor/python.py b/sqlglot/executor/python.py
index e1e597d..3277e65 100644
--- a/sqlglot/executor/python.py
+++ b/sqlglot/executor/python.py
@@ -343,6 +343,9 @@ class PythonExecutor:
         else:
             sink.rows = left.rows + right.rows
 
+        if not math.isinf(step.limit):
+            sink.rows = sink.rows[0 : step.limit]
+
         return self.context({step.name: sink})
 
 
-- 
cgit v1.2.3