def __init__()

in search/mteb/dbpedia/track.py [0:0]

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

    def __init__(self, track, params, **kwargs):
        # choose a suitable index: if there is only one defined for this track
        # choose that one, but let the user always override index
        if len(track.indices) == 1:
            default_index = track.indices[0].name
        else:
            default_index = "_all"

        self._index_name = params.get("index", default_index)
        self._cache = params.get("cache", False)
        self._size = params.get("size", 10)
        self._title_field = params.get("title_field", "title")
        self._text_field = params.get("text_field", "text")
        self._title_boost = params.get("title_boost", 5)
        self._text_boost = params.get("text_boost", 1)
        self._query_file = params.get("query_source", "queries.json")
        self._qrels_file = params.get("qrels_source", "test.tsv")
        self._params = params
        self.infinite = True

        cwd = os.path.dirname(__file__)
        with open(os.path.join(cwd, self._query_file), "r") as file:
            self._queries = json.load(file)
        self._iters = 0

        cwd = os.path.dirname(__file__)
        with open(os.path.join(cwd, self._query_file), "r") as file:
            self._queries = json.load(file)
        self._qrels = read_qrels(os.path.join(cwd, self._qrels_file))