add (multiple) when children are removed because parents are present

alliance-genome · Feb 27, 2019 · b06f61c · b06f61c
1 parent 2f73fa1
commit b06f61c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/genedescriptions/descriptions_generator.py b/genedescriptions/descriptions_generator.py
@@ -262,7 +262,8 @@ def remove_children_if_parents_present(terms, ontology, terms_already_covered: S
                                                                                     term in high_priority_terms):
                 terms_nochildren.append(term)
             elif ancestors_covering_multiple_children is not None:
-                ancestors_covering_multiple_children.update(set(ontology.ancestors(term)).intersection(set(terms)))
+                ancestors_covering_multiple_children.update({ontology.label(term_id, id_if_null=True) for term_id in
+                                                             set(ontology.ancestors(term)).intersection(set(terms))})
         if len(terms_nochildren) < len(terms):
             if terms_already_covered is not None:
                 terms_already_covered.update(set(terms) - set(terms_nochildren))