From 0c159ce842fcdf38fa2977f75c19a2ef55ff259c Mon Sep 17 00:00:00 2001
From: Kyle Barron <kyle@developmentseed.org>
Date: Mon, 16 Dec 2024 16:52:16 -0500
Subject: [PATCH] use await

---
 python/datafusion/record_batch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/datafusion/record_batch.py b/python/datafusion/record_batch.py
index a2ad850c..633e341f 100644
--- 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: