Don't need to cancel when terminating the process, the task should en…

…d on its own
Acly · Oct 14, 2024 · b39458e · b39458e
1 parent 8c16547
commit b39458e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/ai_diffusion/server.py b/ai_diffusion/server.py
@@ -458,11 +458,11 @@ async def stop(self):
             if self._process and self._task:
                 log.info("Stopping server")
                 self._process.terminate()
-                self._task.cancel()
                 await asyncio.wait_for(self._task, timeout=5)
+        except asyncio.CancelledError:
+            pass
         except asyncio.TimeoutError:
             log.warning("Server did not terminate in time")
-            pass
 
     def terminate(self):
         try: