Merge pull request #129 from ic-dev21/main

Fix pour dvd-dev/hilo#282
dvd-dev · Oct 4, 2023 · 2c4af79 · 2c4af79
2 parents 675ea9b + cdbf950
commit 2c4af79
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyhilo/websocket.py b/pyhilo/websocket.py
@@ -298,7 +298,7 @@ async def _clean_queue(self) -> None:
 
     async def async_disconnect(self) -> None:
         """Disconnect from the websocket server."""
-        self._clean_queue()
+        await self._clean_queue()
         if not self.connected:
             return
 
@@ -323,7 +323,7 @@ async def async_listen(self) -> None:
         finally:
             LOG.info("Websocket: Listen completed; cleaning up")
             self._watchdog.cancel()
-            self._clean_queue()
+            await self._clean_queue()
 
             for callback in self._disconnect_callbacks:
                 schedule_callback(callback)