Resolve args.topics -> args.run.topics (#500)

castorini · Apr 24, 2021 · 854c193 · 854c193
1 parent 482aa81
commit 854c193
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyserini/hsearch/__main__.py b/pyserini/hsearch/__main__.py
@@ -145,7 +145,7 @@ def parse_args(parser, commands):
     with open(output_path, 'w') as target_file:
         batch_topics = list()
         batch_topic_ids = list()
-        for index, (topic_id, text) in enumerate(tqdm(list(query_iterator(topics, args.topics)))):
+        for index, (topic_id, text) in enumerate(tqdm(list(query_iterator(topics, args.run.topics)))):
             if args.run.batch_size <= 1 and args.run.threads <= 1:
                 hits = hsearcher.search(text, args.run.hits, args.fusion.alpha)
                 results = [(topic_id, hits)]