X-Git-Url: https://wannabe.guru.org/gitweb/?a=blobdiff_plain;f=remote_worker.py;h=b58c6ba0a66f8d32b2b81af72a66d23493c9b2e5;hb=6cf1d61b80d02937dbca9025c1922568d42a8c73;hp=42aeb854ce633b47e1f1be85b40bce8bf8d436f6;hpb=b454ad295eb3024a238d32bf2aef1ebc3c496b44;p=python_utils.git diff --git a/remote_worker.py b/remote_worker.py index 42aeb85..b58c6ba 100755 --- a/remote_worker.py +++ b/remote_worker.py @@ -6,7 +6,6 @@ results. import logging import os -import platform import signal import threading import sys @@ -32,20 +31,20 @@ cfg.add_argument( type=str, required=True, metavar='FILENAME', - help='The location of the bundle of code to execute.' + help='The location of the bundle of code to execute.', ) cfg.add_argument( '--result_file', type=str, required=True, metavar='FILENAME', - help='The location where we should write the computation results.' + help='The location where we should write the computation results.', ) cfg.add_argument( '--watch_for_cancel', action=argparse_utils.ActionNoYes, - default=False, - help='Should we watch for the cancellation of our parent ssh process?' + default=True, + help='Should we watch for the cancellation of our parent ssh process?', ) @@ -64,7 +63,9 @@ def watch_for_cancel(terminate_event: threading.Event) -> None: saw_sshd = True break if not saw_sshd: - logger.error('Did not see sshd in our ancestors list?! Committing suicide.') + logger.error( + 'Did not see sshd in our ancestors list?! Committing suicide.' + ) os.system('pstree') os.kill(os.getpid(), signal.SIGTERM) time.sleep(5.0)