def main()

in scripts/execute_retrieval.py [0:0]
57 lines of code
12 McCabe index (conditional complexity)

def main(args):

    # load configs
    with open(args.test_config, "r") as fin:
        test_config_json = json.load(fin)

    # create a new directory to log and store results
    log_directory = utils.create_logdir_with_timestamp(args.logdir)
    logger = None

    logger = utils.init_logging(log_directory, args.model_name, logger)
    logger.info("loading {} ...".format(args.model_name))

    if args.model_name == "drqa":
        # DrQA tf-idf
        from kilt.retrievers import DrQA_tfidf

        if args.model_configuration:
            retriever = DrQA_tfidf.DrQA.from_config_file(
                args.model_name, args.model_configuration
            )
        else:
            retriever = DrQA_tfidf.DrQA.from_default_config(args.model_name)
    elif args.model_name == "dpr":
        # DPR
        from kilt.retrievers import DPR_connector

        if args.model_configuration:
            retriever = DPR_connector.DPR.from_config_file(
                args.model_name, args.model_configuration
            )
        else:
            retriever = DPR_connector.DPR.from_default_config(args.model_name)
    elif args.model_name == "dpr_distr":
        # DPR distributed
        from kilt_internal.retrievers import DPR_distr_connector

        if args.model_configuration:
            retriever = DPR_distr_connector.DPR.from_config_file(
                args.model_name, args.model_configuration
            )
        else:
            raise "No default configuration for DPR distributed!"
    elif args.model_name == "blink":
        # BLINK
        from kilt.retrievers import BLINK_connector

        if args.model_configuration:
            retriever = BLINK_connector.BLINK.from_config_file(
                args.model_name, args.model_configuration
            )
        else:
            retriever = BLINK_connector.BLINK.from_default_config(args.model_name)
    elif args.model_name == "bm25":
        # BM25
        from kilt.retrievers import BM25_connector

        if args.model_configuration:
            retriever = BM25_connector.BM25.from_config_file(
                args.model_name, args.model_configuration
            )
        else:
            retriever = BM25_connector.BM25.from_default_config(args.model_name)
    else:
        raise ValueError("unknown retriever model")

    execute(
        logger,
        test_config_json,
        retriever,
        log_directory,
        args.model_name,
        args.output_folder,
    )