a small change to bg task handling

mikeshardmind · Nov 20, 2024 · fd36783 · fd36783
1 parent 2fff43d
commit fd36783
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/async_utils/__init__.py b/async_utils/__init__.py
@@ -13,4 +13,4 @@
 #   limitations under the License.
 
 
-__version__ = "7.0.2"
+__version__ = "7.0.3"
diff --git a/async_utils/bg_tasks.py b/async_utils/bg_tasks.py
@@ -44,7 +44,8 @@ async def __aenter__(self: Self) -> Self:
         return self
 
     async def __aexit__(self, exc_type: type[Exception], exc: Exception, tb: TracebackType):
-        if tsks := self._tasks.copy():
+        while tsks := self._tasks.copy():
             _done, _pending = await asyncio.wait(tsks, timeout=self._exit_timeout)
             for task in _pending:
                 task.cancel()
+            await asyncio.sleep(0)