modify close method according to asyncio docs, resolve TypeError (#112)

numat · Mar 30, 2024 · e0a5160 · e0a5160
1 parent 9dd02b8
commit e0a5160
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/alicat/util.py b/alicat/util.py
@@ -181,7 +181,8 @@ async def _handle_communication(self, command: str) -> Optional[str]:
     async def close(self) -> None:
         """Close the TCP connection."""
         if self.open:
-            await self.connection['writer'].close()
+            self.connection['writer'].close()
+            await self.connection['writer'].wait_closed()
         self.open = False