(self)
| 456 | return self |
| 457 | |
| 458 | async def __anext__(self): |
| 459 | if self.shutdown_event.is_set(): |
| 460 | raise StopAsyncIteration |
| 461 | |
| 462 | try: |
| 463 | item = await self.queue.get() |
| 464 | if item is None: |
| 465 | self.shutdown_event.set() |
| 466 | raise StopAsyncIteration |
| 467 | return item |
| 468 | except asyncio.CancelledError: |
| 469 | self.shutdown_event.set() |
| 470 | raise StopAsyncIteration |
| 471 | |
| 472 | |
| 473 | def _generate_tokens(process_func, *args, **kwargs): |