def generate_rescored_pruned_query()

in msmarco-passage-ranking/track.py [0:0]

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

def generate_rescored_pruned_query(field, query_expansion, num_candidates, boost=1.0):
    return {
        "query": {"sparse_vector": {"field": field, "query_vector": query_expansion, "prune": True, "boost": boost}},
        "rescore": {
            "window_size": num_candidates,
            "query": {
                "rescore_query": {
                    "sparse_vector": {
                        "field": field,
                        "query_vector": query_expansion,
                        "prune": True,
                        "pruning_config": {
                            "only_score_pruned_tokens": True,
                        },
                        "boost": boost,
                    }
                }
            },
        },
    }