Follow generator change

apache · Jul 19, 2023 · 81493b1 · 81493b1
1 parent dba6bb4
commit 81493b1
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/python/pyarrow/ipc.pxi b/python/pyarrow/ipc.pxi
@@ -437,7 +437,11 @@ cdef class MessageReader(_Weakrefable):
 
     def __iter__(self):
         while True:
-            yield self.read_next_message()
+            try:
+                yield self.read_next_message()
+            except StopIteration:
+                # For Cython >= 3.0.0
+                return
 
     def read_next_message(self):
         """
@@ -660,6 +664,7 @@ cdef class RecordBatchReader(_Weakrefable):
             try:
                 yield self.read_next_batch()
             except StopIteration:
+                # For Cython >= 3.0.0
                 return
 
     @property