From 15a896e19d3bb0832ff8543e2ad9a438645f3176 Mon Sep 17 00:00:00 2001 From: BobTheBuidler <70677534+BobTheBuidler@users.noreply.github.com> Date: Tue, 17 Dec 2024 03:23:37 -0400 Subject: [PATCH] feat: rip out contextlib.suppress for speed --- a_sync/task.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/a_sync/task.py b/a_sync/task.py index da42f3d2..6d693563 100644 --- a/a_sync/task.py +++ b/a_sync/task.py @@ -199,11 +199,14 @@ async def _wrapped_set_next( self.__init_loader_coro = exhaust_iterator( self._start_tasks_for_iterables(*iterables), queue=init_loader_queue ) - with contextlib.suppress(_NoRunningLoop): - # its okay if we get this exception, we can start the task as soon as the loop starts - self._init_loader self._init_loader_next = init_loader_queue.get_all + try: + self._init_loader + except _NoRunningLoop: + # its okay if we get this exception, we can start the task as soon as the loop starts + pass + def __repr__(self) -> str: return f"<{type(self).__name__} for {self._wrapped_func} kwargs={self._wrapped_func_kwargs} tasks={len(self)} at {hex(id(self))}>"