From da1842e50c8381917fab6e44b600c82c58cafa26 Mon Sep 17 00:00:00 2001 From: deeppppp Date: Sun, 3 Nov 2019 16:26:06 -0600 Subject: [PATCH] Bug fix --- argv/checkpoints.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/argv/checkpoints.py b/argv/checkpoints.py index 29a74ac..9321f72 100644 --- a/argv/checkpoints.py +++ b/argv/checkpoints.py @@ -45,12 +45,13 @@ def check_args_checkpoints_parser(parser, args): def check_arg_checkpoints(parser, args): Conf.log.debug(args.checkpoints) - for _, v in args.checkpoints.items(): - if not os.path.isfile(v): - parser.error( - "Checkpoints file not found. " - "You can download them using : {} checkpoints download".format(sys.argv[0]) - ) + if not ('download' in str(args.func)): + for _, v in args.checkpoints.items(): + if not os.path.isfile(v): + parser.error( + "Checkpoints file not found! " + "You can download them using : {} checkpoints download".format(sys.argv[0]) + ) def set_arg_checkpoints(args):