Spaces:

orionweller
/

retrieval-prompting

Sleeping

orionweller commited on Sep 9, 2024

Commit

f7300ce

1 Parent(s): c74038e

sort?

Files changed (1) hide show

app.py CHANGED Viewed

@@ -197,7 +197,7 @@ def load_queries(dataset_name):
     dataset = ir_datasets.load(f"beir/{dataset_name.lower()}" + ("/test" if dataset_name == "scifact" else ""))
     queries[dataset_name] = []
-    query2qid[dataset_name] = {}
     q_lookups[dataset_name] = {}
     qrels[dataset_name] = {}
     for query in dataset.queries_iter():

     dataset = ir_datasets.load(f"beir/{dataset_name.lower()}" + ("/test" if dataset_name == "scifact" else ""))
     queries[dataset_name] = []
+    query2qid[dataset_name] = defaultdict(dict)
     q_lookups[dataset_name] = {}
     qrels[dataset_name] = {}
     for query in dataset.queries_iter():