def get_parser()

in dpr_scale/run_retrieval_fb.py [0:0]

• 34 lines of code
• 1 McCabe index (conditional complexity)

def get_parser():
    parser = argparse.ArgumentParser()
    parser.add_argument(
        "--ctx_embeddings_dir",
        type=str,
        default="",
    )
    parser.add_argument(
        "--query_emb_path",
        type=str,
        default="",
        help="if left empty, will use <ctx_embeddings_dir>/query_reps.pkl"
    )
    parser.add_argument(
        "--questions_jsonl_path",
        type=str,
        default="",
    )
    parser.add_argument(
        "--passages_tsv_path",
        type=str,
        default="",
    )
    parser.add_argument(
        "--output_json_path",
        type=str,
        default="",
    )
    parser.add_argument(
        "--topk",
        type=int,
        default=100,
    )
    return parser