Add ChemicalSubstance to SmallMolecule queries

ranking-agent · Aug 6, 2021 · cd788d6 · cd788d6
1 parent 981b1b4
commit cd788d6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/strider/trapi.py b/strider/trapi.py
@@ -326,6 +326,8 @@ async def fill_categories_predicates(
     for node in qg['nodes'].values():
         if ('categories' not in node) or (node['categories'] is None):
             node['categories'] = ['biolink:NamedThing']
+        elif "biolink:ChemicalSubstance" in node["categories"]:
+            node["categories"].append("biolink:SmallMolecule")
 
     logger.debug("Contacting node normalizer to get categorys for curies")