fix: more type checking client.py

tutorintelligence · May 1, 2024 · 211ee97 · 211ee97
1 parent 99a0460
commit 211ee97
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tcp_modbus_aio/client.py b/tcp_modbus_aio/client.py
@@ -200,14 +200,14 @@ async def _get_tcp_connection(
 
         return reader, writer
 
-    async def __aenter__(self) -> Self:
+    async def __aenter__(self) -> "Self":
         return self
 
     async def __aexit__(
         self,
         exc_type: type[BaseException] | None,
         exc_val: BaseException | None,
-        exc_tb: TracebackType | None,
+        exc_tb: "TracebackType" | None,
     ) -> None:
         await self.close()