Skip to content

Commit e334220

Browse files
authored
Speed up query list generation for dense matching (#453)
1 parent d0e8494 commit e334220

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hloc/match_dense.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -485,7 +485,7 @@ def match_and_assign(
485485
pairs = parse_retrieval(pairs_path)
486486
pairs = [(q, r) for q, rs in pairs.items() for r in rs]
487487
pairs = find_unique_new_pairs(pairs, None if overwrite else match_path)
488-
required_queries = set(sum(pairs, ()))
488+
required_queries = set(chain.from_iterable(pairs))
489489

490490
name2ref = {
491491
n: i for i, p in enumerate(feature_paths_refs) for n in list_h5_names(p)

0 commit comments

Comments
 (0)