From 40e9bff0aece1cc08079c45fff67c9ff9f94be6f Mon Sep 17 00:00:00 2001
From: Scott Donnelly <scott@donnel.ly>
Date: Sat, 17 Aug 2024 00:20:42 +0100
Subject: [PATCH] fix(row selection): off-by-one error

---
 crates/iceberg/src/arrow/reader.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/crates/iceberg/src/arrow/reader.rs b/crates/iceberg/src/arrow/reader.rs
index 7876b5ba1..7d7017569 100644
--- a/crates/iceberg/src/arrow/reader.rs
+++ b/crates/iceberg/src/arrow/reader.rs
@@ -441,7 +441,7 @@ impl ArrowReader {
                 // skip row groups that aren't present in selected_row_groups
                 if idx == selected_row_groups[selected_row_groups_idx] {
                     selected_row_groups_idx += 1;
-                    if selected_row_groups_idx == selected_row_groups.len() {
+                    if selected_row_groups_idx == selected_row_groups.len() + 1 {
                         break;
                     }
                 } else {