Add type hints to the main module

This commit is contained in:
coolneng 2021-07-05 03:51:53 +02:00
parent bcc4f4b4d4
commit 72e3de945a
Signed by: coolneng
GPG Key ID: 9893DA236405AF57
1 changed files with 3 additions and 3 deletions

View File

@ -1,10 +1,10 @@
from argparse import ArgumentParser from argparse import ArgumentParser, Namespace
from time import time from time import time
from model import run from model import run
def parse_arguments(): def parse_arguments() -> Namespace:
parser = ArgumentParser() parser = ArgumentParser()
parser.add_argument( parser.add_argument(
"data_file", help="FASTQ file containing the sequences with errors" "data_file", help="FASTQ file containing the sequences with errors"
@ -15,7 +15,7 @@ def parse_arguments():
return parser.parse_args() return parser.parse_args()
def main(): def main() -> None:
args = parse_arguments() args = parse_arguments()
start_time = time() start_time = time()
run(data_file=args.data_file, label_file=args.label_file) run(data_file=args.data_file, label_file=args.label_file)