Shortened code for execute action

Signed-off-by: Jim Martens <github@2martens.de>
This commit is contained in:
Jim Martens 2019-09-02 13:19:00 +02:00
parent d58179fa2a
commit 6e85ab86eb
1 changed files with 3 additions and 34 deletions

View File

@ -47,15 +47,7 @@ def main() -> None:
_build_measure(sub_parsers[7])
args = _get_user_input(parser)
_execute_action(args,
cli.config,
cli.prepare,
cli.train,
cli.test,
cli.evaluate,
cli.visualise,
cli.visualise_metrics,
cli.measure_mapping)
_execute_action(args)
def _build_general(parser: argparse.ArgumentParser) -> None:
@ -96,31 +88,8 @@ def _get_action(component: str) -> callable:
return getattr(cli, component)
def _execute_action(args: argparse.Namespace,
on_config: callable,
on_prepare: callable,
on_train: callable,
on_test: callable,
on_evaluate: callable,
on_visualise: callable,
on_visualise_metrics: callable,
on_measure: callable) -> None:
if args.component == "config":
on_config(args)
elif args.component == "prepare":
on_prepare(args)
elif args.component == "train":
on_train(args)
elif args.component == "test":
on_test(args)
elif args.component == "evaluate":
on_evaluate(args)
elif args.component == "visualise":
on_visualise(args)
elif args.component == "visualise_metrics":
on_visualise_metrics(args)
elif args.component == "measure_mapping":
on_measure(args)
def _execute_action(args: argparse.Namespace) -> None:
getattr(cli, args.component)(args)
def _build_config(parser: argparse.ArgumentParser) -> None: