def params()

in msmarco-v2-vector/track.py [0:0]

• 22 lines of code
• 4 McCabe index (conditional complexity)

    def params(self):
        top_k = self._params.get("k", 10)
        num_candidates = self._params.get("num-candidates", 50)
        num_rescore = self._params.get("num-rescore", 0)
        query_vec = self._queries[self._iters]
        result = {
            "index": self._index_name,
            "cache": self._params.get("cache", False),
            "size": top_k,
            "body": {
                "knn": {"field": "emb", "query_vector": query_vec, "k": max(top_k, num_rescore), "num_candidates": num_candidates},
                "_source": False,
            },
        }
        if num_rescore > 0:
            result["body"]["rescore"] = get_rescore_query(query_vec, num_rescore)
        if "filter" in self._params:
            result["body"]["knn"]["filter"] = self._params["filter"]

        self._iters += 1
        if self._iters >= self._maxIters:
            self._iters = 0
        return result