updates to add type hinting to uses of Callable

daemon/core/emulator/distributed.py

@@ -21,7 +21,6 @@ from core.nodes.network import CoreNetwork, CtrlNet
 if TYPE_CHECKING:
     from core.emulator.session import Session
 
-
 LOCK = threading.Lock()
 CMD_HIDE = True
 
@@ -139,7 +138,7 @@ class DistributedController:
         cmd = f"mkdir -p {self.session.session_dir}"
         server.remote_cmd(cmd)
 
-    def execute(self, func: Callable) -> None:
+    def execute(self, func: Callable[[DistributedServer], None]) -> None:
         """
         Convenience for executing logic against all distributed servers.
 

daemon/core/emulator/session.py

@@ -1206,7 +1206,7 @@ class Session:
                     ExceptionLevels.ERROR, "Session.run_state_hooks", None, message
                 )
 
-    def add_state_hook(self, state: int, hook: Callable) -> None:
+    def add_state_hook(self, state: int, hook: Callable[[int], None]) -> None:
         """
         Add a state hook.
 
@@ -1222,7 +1222,7 @@ class Session:
         if self.state == state:
             hook(state)
 
-    def del_state_hook(self, state: int, hook: Callable) -> None:
+    def del_state_hook(self, state: int, hook: Callable[[int], None]) -> None:
         """
         Delete a state hook.