Attempt to fix unit test again

vkottler · Feb 16, 2023 · ff4d1ab · ff4d1ab
1 parent c98be76
commit ff4d1ab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/runtimepy/task/basic/__init__.py b/runtimepy/task/basic/__init__.py
@@ -8,6 +8,7 @@
 from abc import ABC as _ABC
 from abc import abstractmethod as _abstractmethod
 import asyncio as _asyncio
+from contextlib import suppress as _suppress
 from logging import getLogger as _getLogger
 from typing import Optional as _Optional
 
@@ -72,7 +73,8 @@ async def task(self, period_s: float) -> _asyncio.Task[None]:
         if self._task is not None:
             if not self._task.done():
                 self._task.cancel()
-                await self._task
+                with _suppress(_asyncio.CancelledError):
+                    await self._task
                 _log_task_exception(self._task, logger=self.logger)
 
             self._task = None