load takes an Iterator

BritishGeologicalSurvey · Sep 13, 2023 · 76d424e · 76d424e
1 parent 88a45ea
commit 76d424e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/etlhelper/etl.py b/etlhelper/etl.py
@@ -504,7 +504,7 @@ def copy_table_rows(
 def load(
         table: str,
         conn: Connection,
-        rows: list,
+        rows: Iterator,
         transform: Optional[Callable[[Chunk], Chunk]] = None,
         on_error: Optional[Callable] = None,
         commit_chunks: bool = True,