Try to fix minor issues

SpiNNakerManchester · Nov 16, 2024 · 2d996c4 · 2d996c4
1 parent 3cae3f1
commit 2d996c4
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 2 deletions.
diff --git a/spinn_front_end_common/utilities/connections/live_event_connection.py b/spinn_front_end_common/utilities/connections/live_event_connection.py
@@ -363,8 +363,11 @@ def __read_database_callback(self, db_reader: DatabaseReader):
 
         for label, vertex_size in vertex_sizes.items():
             for init_callback in self.__init_callbacks[label]:
+                logger.info("LiveEventConnection: calling init callback {}",
+                            init_callback)
                 init_callback(
                     label, vertex_size, run_time_ms, machine_timestep / 1000.0)
+        logger.info("LiveEventConnection: database read")
 
     def __init_sender(
             self, database: DatabaseReader, vertex_sizes: Dict[str, int]):

diff --git a/spinn_front_end_common/utilities/database/database_connection.py b/spinn_front_end_common/utilities/database/database_connection.py
@@ -144,6 +144,7 @@ def __read_db(self, toolchain_address: Tuple[str, int], data: bytes):
             # Call the callback
             with DatabaseReader(database_path) as db_reader:
                 for db_callback in self.__database_callbacks:
+                    logger.info("Calling database callback {}", db_callback)
                     db_callback(db_reader)
         else:
             logger.warning("Database path was empty - assuming no database")

diff --git a/spinn_front_end_common/utilities/notification_protocol/notification_protocol.py b/spinn_front_end_common/utilities/notification_protocol/notification_protocol.py
@@ -96,8 +96,6 @@ def send_start_resume_notification(self) -> None:
         """
         logger.info("** Sending start / resume message to external sources "
                     "to state the simulation has started or resumed. **")
-        if self.__wait_for_read_confirmation:
-            self.wait_for_confirmation()
         eieio_command_message = NotificationProtocolStartResume()
         for c in self.__database_message_connections:
             try:
@@ -181,6 +179,9 @@ def __do_read_notify(self, database_path: Optional[str]) -> None:
         if self.__wait_for_read_confirmation:
             for c in self.__database_message_connections:
                 try:
+                    logger.info(
+                        "Waiting for confirmation from {}:{}",
+                        c.remote_ip_address, c.remote_port)
                     c.receive_eieio_message()
                     logger.info(
                         "** Confirmation from {}:{} received, continuing **",