Update launcher.py

Author: apoorvkh

src/torchrunx/launcher.py

@@ -181,7 +181,6 @@ def launch(
                             else:
                                 e += f"{v.message['message']}\n"
                                 e += f"{v.message['extraInfo']['py_callstack']}\n\n"
-                print_process.kill()
                 raise RuntimeError(e)
     except:
         # kill all agents
@@ -191,9 +190,9 @@ def launch(
                 hostname=agent_hostname,
                 ssh_config_file=ssh_config_file,
             )
-        print_process.kill()   
         raise
+    finally:
+        print_process.kill()
 
-    print_process.kill()  # TODO: or close?
     return_values: dict[int, Any] = dict(ChainMap(*[s.return_values for s in agent_statuses]))
     return return_values