Skip to content

Commit 2993854

Browse files
committed
fix: Update loop condition in collect_record_batches_to_display for correct row handling
1 parent b467100 commit 2993854

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/dataframe.rs

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -905,7 +905,7 @@ async fn collect_record_batches_to_display(
905905
let mut record_batches = Vec::default();
906906
let mut has_more = false;
907907

908-
while size_estimate_so_far < max_bytes && rows_so_far < max_rows && rows_so_far < min_rows {
908+
while (size_estimate_so_far < max_bytes && rows_so_far < max_rows) || rows_so_far < min_rows {
909909
let mut rb = match stream.next().await {
910910
None => {
911911
break;

0 commit comments

Comments
 (0)