multimap resolving fixed

ablab · Sep 13, 2024 · e3f3524 · e3f3524
1 parent 1786f0a
commit e3f3524
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/dataset_processor.py b/src/dataset_processor.py
@@ -624,7 +624,7 @@ def resolve_multimappers(self, chr_ids, sample, multimapped_reads):
 
         for assignment_list in multimapped_reads.values():
             if len(assignment_list) > 1:
-                multimap_resolver.resolve(assignment_list)
+                assignment_list = multimap_resolver.resolve(assignment_list)
                 resolved_lists = defaultdict(list)
                 for a in assignment_list:
                     resolved_lists[a.chr_id].append(a)

diff --git a/src/multimap_resolver.py b/src/multimap_resolver.py
@@ -215,5 +215,5 @@ def select_noninformative(assignment_list, assignment_indices):
                 best_assignment = info[2]
 
         assert best_assignment != -1
-        return [assignment_list[best_assignment]]
+        return MultimapResolver.filter_assignments(assignment_list, {best_assignment})