style/doc fixes (NFC)

This commit is contained in:
Sébastien Bourdeauducq 2024-02-27 11:10:21 +08:00
parent b2ba087acd
commit bafa69098a
3 changed files with 11 additions and 10 deletions

View File

@ -57,10 +57,10 @@ def get_argparser():
log_args(parser) log_args(parser)
parser.add_argument("--name", parser.add_argument("--name",
help="friendly name, displayed in dashboards " help="friendly name, displayed in dashboards "
"to identify master instead of server address") "to identify master instead of server address")
parser.add_argument("--log-submissions", default=None, parser.add_argument("--log-submissions", default=None,
help="set the filename to create the experiment subimission") help="log experiment submissions to specified file")
return parser return parser
@ -81,8 +81,7 @@ def main():
bind, args.port_broadcast)) bind, args.port_broadcast))
atexit_register_coroutine(server_broadcast.stop, loop=loop) atexit_register_coroutine(server_broadcast.stop, loop=loop)
log_forwarder.callback = (lambda msg: log_forwarder.callback = lambda msg: server_broadcast.broadcast("log", msg)
server_broadcast.broadcast("log", msg))
def ccb_issue(service, *args, **kwargs): def ccb_issue(service, *args, **kwargs):
msg = { msg = {
"service": service, "service": service,
@ -106,7 +105,8 @@ def main():
repo_backend, worker_handlers, args.experiment_subdir) repo_backend, worker_handlers, args.experiment_subdir)
atexit.register(experiment_db.close) atexit.register(experiment_db.close)
scheduler = Scheduler(RIDCounter(), worker_handlers, experiment_db, args.log_submissions) scheduler = Scheduler(RIDCounter(), worker_handlers, experiment_db,
args.log_submissions)
scheduler.start(loop=loop) scheduler.start(loop=loop)
atexit_register_coroutine(scheduler.stop, loop=loop) atexit_register_coroutine(scheduler.stop, loop=loop)

View File

@ -178,7 +178,8 @@ class ArgumentManager(ProcessArgumentManager):
user_input_deser = pyon.decode(user_input) user_input_deser = pyon.decode(user_input)
value = processor.process(user_input_deser) value = processor.process(user_input_deser)
except: except:
logger.error("failed to process user input, retrying", exc_info=True) logger.error("failed to process user input, retrying",
exc_info=True)
else: else:
success = True success = True
result[key] = value result[key] = value

View File

@ -156,7 +156,7 @@ class RunPool:
run = Run(rid, pipeline_name, wd, expid, priority, due_date, flush, run = Run(rid, pipeline_name, wd, expid, priority, due_date, flush,
self, repo_msg=repo_msg) self, repo_msg=repo_msg)
if self.log_submissions is not None: if self.log_submissions is not None:
self.log_submission(rid, expid) self.log_submission(rid, expid)
self.runs[rid] = run self.runs[rid] = run
self.state_changed.notify() self.state_changed.notify()
@ -514,4 +514,4 @@ class Scheduler:
if run.termination_requested: if run.termination_requested:
return True return True
return False return False