Improve asyncio.wait function to create the futures set just one time.