Fighting with type hints.

diff --git a/state_tracker.py b/state_tracker.py
index b375f8928dc09c6f359d6e99846d06f1db947c5e..1cf62af6ea8000e5d1ed32646ec0a212a0b55faf 100644 (file)
--- a/state_tracker.py
+++ b/state_tracker.py
@@ -109,7 +109,7 @@ class AutomaticStateTracker(StateTracker):
     """
 
     @background_thread
-    def pace_maker(self, should_terminate) -> None:
+    def pace_maker(self, should_terminate: threading.Event) -> None:
         """Entry point for a background thread to own calling heartbeat()
         at regular intervals so that the main thread doesn't need to do
         so.

diff --git a/thread_utils.py b/thread_utils.py
index 4db4cf68b4ef916f323f90bae492d1c59dca361f..2375f3df3938b6a5562911af28202a909b0a0cd7 100644 (file)
--- a/thread_utils.py
+++ b/thread_utils.py
@@ -6,7 +6,7 @@ import functools
 import logging
 import os
 import threading
-from typing import Callable, Optional, Tuple
+from typing import Any, Callable, Optional, Tuple
 
 # This module is commonly used by others in here and should avoid
 # taking any unnecessary dependencies back on them.
@@ -62,7 +62,7 @@ def is_current_thread_main_thread() -> bool:
 
 
 def background_thread(
-    _funct: Optional[Callable],
+    _funct: Optional[Callable[..., Any]],
 ) -> Callable[..., Tuple[threading.Thread, threading.Event]]:
     """A function decorator to create a background thread.