fix: handle TYPE_CHECKING correctly

tutorintelligence · May 1, 2024 · f1dcac6 · f1dcac6
1 parent 40a3e23
commit f1dcac6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tcp_modbus_aio/client.py b/tcp_modbus_aio/client.py
@@ -30,7 +30,7 @@
 
 @dataclass
 class CoilWatchStatus:
-    msg: ModbusFunction
+    msg: "ModbusFunction"
 
     memo_key: Any
     expiry: float
@@ -228,7 +228,7 @@ async def close(self) -> None:
             self._ping_loop = None
 
     def log_watch(
-        self, msg: ModbusFunction, *, memo_key: Any, period: float, hz: float
+        self, msg: "ModbusFunction", *, memo_key: Any, period: float, hz: float
     ) -> None:
         """
         Triggers a loop that reads the coil(s) at the given index at the given frequency and