diff --git a/ir_datasets/formats/base.py b/ir_datasets/formats/base.py index 301e63f..8d4f03c 100644 --- a/ir_datasets/formats/base.py +++ b/ir_datasets/formats/base.py @@ -185,7 +185,7 @@ def qlogs_handler(self): return self -BaseQueries.EXTENSIONS['queries_dict'] = lambda x: {q.query_id: q for q in x.iter_queries()} +BaseQueries.EXTENSIONS['queries_dict'] = lambda x: {q.query_id: q for q in x.queries_iter()} def qrels_dict(qrels_handler):