Update transactron/testing/testbenchio.py

Co-authored-by: Arusekk <[email protected]>
kuznia-rdzeni · Nov 18, 2024 · 4252935 · 4252935
1 parent c14ff5c
commit 4252935
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/transactron/testing/testbenchio.py b/transactron/testing/testbenchio.py
@@ -36,9 +36,7 @@ def call(self, tbio: "TestbenchIO", data: dict[str, Any] = {}, /, **kwdata):
         return CallTrigger(self.sim, (*self.calls_and_values, (tbio, data or kwdata)))
 
     async def until_done(self) -> Any:
-        call = self.__aiter__()
-        while True:
-            results = await call.__anext__()
+        async for results in self:
             if any(res is not None for res in results):
                 return results