fix: update RecordBatchStream methods to return pa.RecordBatch instances

Author: kosiew

python/datafusion/record_batch.py

@@ -60,17 +60,17 @@ def __init__(self, record_batch_stream: df_internal.RecordBatchStream) -> None:
         """This constructor is typically not called by the end user."""
         self.rbs = record_batch_stream
 
-    def next(self) -> RecordBatch:
-        """See :py:func:`__next__` for the iterator function."""
+    def next(self) -> pa.RecordBatch:
+        """Retrieve the next :py:class:`pa.RecordBatch`."""
         return next(self)
 
-    async def __anext__(self) -> RecordBatch:
-        """Async iterator function."""
+    async def __anext__(self) -> pa.RecordBatch:
+        """Async iterator returning :py:class:`pa.RecordBatch`."""
         next_batch = await self.rbs.__anext__()
-        return RecordBatch(next_batch)
+        return next_batch.to_pyarrow()
 
     def __next__(self) -> pa.RecordBatch:
-        """Iterator function."""
+        """Iterator returning :py:class:`pa.RecordBatch`."""
         next_batch = next(self.rbs)
         return next_batch.to_pyarrow()