From 5c9e2f47128199701985d381211504e965a014f5 Mon Sep 17 00:00:00 2001 From: coolneng Date: Tue, 6 Jul 2021 18:56:50 +0200 Subject: [PATCH] Add the async CLI execution of the inference --- src/main.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/src/main.py b/src/main.py index 0ef6fd7..387cbb5 100644 --- a/src/main.py +++ b/src/main.py @@ -1,3 +1,4 @@ +from asyncio import run from argparse import ArgumentParser, Namespace from model import infer_sequence, train_model @@ -18,17 +19,17 @@ def parse_arguments() -> Namespace: return parser.parse_args() -def execute_task(args): +async def execute_task(args): if args.task == "train": train_model(data_file=args.data_file, label_file=args.label_file) else: - prediction = infer_sequence(sequence=args.sequence) + prediction = await infer_sequence(sequence=args.sequence) print(f"Error-corrected sequence: {prediction}") def main() -> None: args = parse_arguments() - execute_task(args) + run(execute_task(args)) if __name__ == "__main__":