use await

kylebarron · kylebarron · commit 0c159ce842fc · 2024-12-16T16:52:16.000-05:00
diff --git a/python/datafusion/record_batch.py b/python/datafusion/record_batch.py
@@ -63,7 +63,7 @@ def next(self) -> RecordBatch | None:
 
     async def __anext__(self) -> RecordBatch:
         """Async iterator function."""
-        next_batch = self.rbs.__anext__()
+        next_batch = await self.rbs.__anext__()
         return RecordBatch(next_batch)
 
     def __next__(self) -> RecordBatch:
