use await

apache · Dec 16, 2024 · 0c159ce · 0c159ce
1 parent 4862d9c
commit 0c159ce
Showing 1 changed file with 1 addition and 1 deletion.
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: