Fix iterator termination

* Check error after reading each shard

1 files changed, 4 insertions, 4 deletions

diff --git a/dataset/parquet.go b/dataset/parquet.go
index 4b3d858..81454f8 100644
--- a/dataset/parquet.go
+++ b/dataset/parquet.go
@@ -72,16 +72,16 @@ func (p *ParquetReader) Texts() iter.Seq2[int, string] {
 
 		for _, name := range p.shards {
 			for text := range p.read(name) {
-				if p.err != nil {
-					return
-				}
-
 				if !yield(n, text) {
 					return
 				}
 
 				n++
 			}
+
+			if p.err != nil {
+				return
+			}
 		}
 	}
 }