Fighting with type hints.

2 files changed, 3 insertions, 3 deletions

diff --git a/state_tracker.py b/state_tracker.py
index b375f89..1cf62af 100644
--- 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 4db4cf6..2375f3d 100644
--- 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.