INNER CODE UNIT · Python
parse_args
AMontgomerie/question_generator · run_qg.py:6
def parse_args() -> argparse.Namespace:
parser = argparse.ArgumentParser()
parser.add_argument(
"--answer_style",
default="all",
type=str,
help="The desired type of answers. Choose from ['all', 'sentences', 'multiple_choice']",
)
parser.add_argument("--model_dir", type=str, default=None)
parser.add_argument("--num_questions", type=int, default=10)
parser.add_argument("--show_answers", dest="show_answers", action="store_true", default=True)
parser.add_argument("--text_file", type=str, required=True)
parser.add_argument("--use_qa_eval", dest="use_qa_eval", action="store_true", default=True)
return parser.parse_args()
if __name__ == "__main__":
args = parse_args()