FIX/#576

mabel-dev · Oct 3, 2022 · 882eba0 · 882eba0
1 parent 2d1132b
commit 882eba0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/opteryx/utils/__init__.py b/opteryx/utils/__init__.py
@@ -46,7 +46,7 @@ def peak(generator):  # type:ignore
         item = next(generator)
     except StopIteration:
         return None, []
-    return item, itertools.chain(item, generator)
+    return item, itertools.chain([item], generator)
 
 
 def fuzzy_search(name, candidates):

diff --git a/opteryx/utils/arrow.py b/opteryx/utils/arrow.py
@@ -161,7 +161,6 @@ def limit_records(data_pages, limit):
 
     # if we don't actually have a limit set, just return
     if limit is None:
-        data_pages = list(data_pages)
         return pyarrow.concat_tables(data_pages, promote=True)
 
     for page in data_pages: