mypy fixes

SpiNNakerManchester · Jul 23, 2024 · 856dfc4 · 856dfc4
1 parent 7965f95
commit 856dfc4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/spinnman/spalloc/spalloc_client.py b/spinnman/spalloc/spalloc_client.py
@@ -145,6 +145,7 @@ def get_job(self, job_id: str) -> SpallocJob:
         :param str job_id: The job id.
         :rtype: SpallocJob
         """
+        assert self.__session
         return _SpallocJob(
             self.__session, fix_url(f"{self.__jobs_url}/{job_id}"))
 

diff --git a/spinnman/utilities/locate_connected_machine_ip_address.py b/spinnman/utilities/locate_connected_machine_ip_address.py
@@ -20,7 +20,7 @@
 from spinnman.connections.udp_packet_connections import IPAddressesConnection
 
 
-def locate_connected_machine(handler: Callable[[str, float], None]):
+def locate_connected_machine(handler: Callable[[str, float], bool]):
     """
     Locates any SpiNNaker machines IP addresses from the auto-transmitted
     packets from non-booted SpiNNaker machines.
@@ -51,7 +51,7 @@ def _ctrlc_handler(sig, frame):
         print("Exiting")
         sys.exit()
 
-    def _print_connected(ip_address: str, timestamp: float):
+    def _print_connected(ip_address: str, timestamp: float) -> bool:
         try:
             hostname = f" ({socket.gethostbyaddr(ip_address)[0]})"
         except Exception:  # pylint: disable=broad-except