Fixes encoding error when opening OpenResearch's queries file (#1239)

castorini · May 28, 2020 · d55531a · d55531a
1 parent a6a968a
commit d55531a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/python/openresearch/retrieve.py b/src/main/python/openresearch/retrieve.py
@@ -57,7 +57,7 @@
 
     with open(args.output, 'w') as fout:
         start_time = time.time()
-        for line_number, line in tqdm(enumerate(open(args.qid_queries))):
+        for line_number, line in tqdm(enumerate(open(args.qid_queries, encoding='utf-8'))):
             query_id, query = line.strip().split('\t')
             # We return one more result because it is almost certain that we will
             # retrieve the document that originated the query.