include terms with no 'type'

alliance-genome · Oct 2, 2019 · b87e1fa · b87e1fa
1 parent c4e3482
commit b87e1fa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/genedescriptions/ontology_tools.py b/genedescriptions/ontology_tools.py
@@ -11,7 +11,7 @@
 
 def set_all_depths(ontology: Ontology, relations: List[str] = None, comparison_func=max):
     for root_id in ontology.get_roots():
-        if ontology.node_type(root_id) == "CLASS":
+        if "type" not in ontology.node(root_id) or ontology.node_type(root_id) == "CLASS":
             set_all_depths_in_subgraph(ontology=ontology, root_id=root_id, relations=relations,
                                        comparison_func=comparison_func)