From 1356cdc2cd62d0d8b9ebe4201b190d3919f84835 Mon Sep 17 00:00:00 2001 From: Seongbeom Park Date: Sun, 28 Apr 2024 05:11:18 +0900 Subject: [PATCH] execute single problem --- run.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/run.py b/run.py index e53f1e4..37548e8 100644 --- a/run.py +++ b/run.py @@ -14,12 +14,11 @@ problems = [ if __name__ == '__main__': parser = argparse.ArgumentParser() - parser.add_argument('--problems', type=list, + parser.add_argument('-p', '--problem', type=str, choices=problems, - default=problems[-1:]) + default=problems[-1]) args = parser.parse_args() - for problem in args.problems: - print(problem) + print(args.problem) - sys.path.append(os.path.join(os.getcwd(), problem)) - __import__(problem, fromlist=['test']) + sys.path.append(os.path.join(os.getcwd(), args.problem)) + __import__(args.problem, fromlist=['test'])