Skip to content

Commit

Permalink
fix(python): Address issue with read_database draining iter_batches…
Browse files Browse the repository at this point in the history
… early (#15504)
  • Loading branch information
alexander-beedie authored Apr 6, 2024
1 parent c9da392 commit c7910b2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion py-polars/polars/io/database/_executor.py
Original file line number Diff line number Diff line change
Expand Up @@ -276,7 +276,7 @@ def _from_rows(
orient="row",
)
for rows in (
list(self._fetchmany_rows(self.result, batch_size))
self._fetchmany_rows(self.result, batch_size)
if iter_batches
else [self._fetchall_rows(self.result)] # type: ignore[list-item]
)
Expand Down

0 comments on commit c7910b2

Please sign in to comment.