Apply suggestions from code review

Co-authored-by: Iaroslav Igoshev <[email protected]>

unidist/core/backends/mpi/core/controller/api.py

@@ -261,7 +261,7 @@ def init():
         atexit.register(_termination_handler)
         signal.signal(signal.SIGTERM, _termination_handler)
         signal.signal(signal.SIGINT, _termination_handler)
-        # Exit the init function in root only after monitor and worker loops have started.
+        # Exit the init function in root only after monitor and worker loops have started
         mpi_state.comm.Barrier()
         return
     elif mpi_state.is_monitor_process():

unidist/core/backends/mpi/core/monitor/loop.py

@@ -183,7 +183,7 @@ def monitor_loop():
     # Once all workers excluding ``Root`` and ``Monitor`` ranks are ready to shutdown,
     # ``Monitor` sends the shutdown signal to every worker, as well as notifies ``Root`` that
     # it can exit the program.
-    # Barrier in monitor process to check if monitor loop has started
+    # Barrier to check if monitor process is ready to start the communication loop
     mpi_state.comm.Barrier()
     while True:
         # Listen receive operation from any source

unidist/core/backends/mpi/core/worker/loop.py

@@ -96,7 +96,7 @@ async def worker_loop():
     # Once all workers excluding ``Root`` and ``Monitor`` ranks are ready to shutdown,
     # ``Monitor` sends the shutdown signal to every worker so they can exit the loop.
 
-    # Barrier in worker process to check if worker loop has started
+    # Barrier to check if worker process is ready to start the communication loop
     mpi_state.comm.Barrier()
     while True:
         # Listen receive operation from any source