#!/usr/bin/env python3.7 def f(): return (i*2 async for i in arange(42)) def g(): return (something_long * something_long async for something_long in async_generator(with_an_argument)) async def func(): if test: out_batched = [ i async for i in aitertools._async_map( self.async_inc, arange(8), batch_size=3 ) ] # output #!/usr/bin/env python3.7 def f(): return (i * 2 async for i in arange(42)) def g(): return ( something_long * something_long async for something_long in async_generator(with_an_argument) ) async def func(): if test: out_batched = [ i async for i in aitertools._async_map( self.async_inc, arange(8), batch_size=3 ) ]