Expected:
async def func():
if test:
out_batched = [
i
async for i in aitertools._async_map(
self.async_inc, arange(8), batch_size=3
)
]
Actual:
async def func():
if test:
out_batched = [
i async
for i in aitertools._async_map(self.async_inc, arange(8), batch_size=3)
]