From 28dd04190e0802c8b1ae4dc7ad0e5a43d6439bba Mon Sep 17 00:00:00 2001 From: coolneng Date: Mon, 5 Jul 2021 03:51:53 +0200 Subject: [PATCH] Add typehints to main module --- src/main.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/main.py b/src/main.py index 1015d87..0fa69ef 100644 --- a/src/main.py +++ b/src/main.py @@ -1,10 +1,10 @@ -from argparse import ArgumentParser +from argparse import ArgumentParser, Namespace from time import time from model import run -def parse_arguments(): +def parse_arguments() -> Namespace: parser = ArgumentParser() parser.add_argument( "data_file", help="FASTQ file containing the sequences with errors" @@ -15,7 +15,7 @@ def parse_arguments(): return parser.parse_args() -def main(): +def main() -> None: args = parse_arguments() start_time = time() run(data_file=args.data_file, label_file=args.label_file)