Fixing Tests

RTXteam · Aug 6, 2024 · 030c38a · 030c38a
1 parent f8eecfb
commit 030c38a
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/code/ARAX/ARAXQuery/Infer/scripts/infer_utilities.py b/code/ARAX/ARAXQuery/Infer/scripts/infer_utilities.py
@@ -194,6 +194,8 @@ def genrete_treat_subgraphs(self, response: ARAXResponse, top_drugs: pd.DataFram
                     node_info = xdtdmapping.get_node_info(node_id=drug_canonical_id)
                 except:
                     continue
+                if not node_info:
+                    continue
                 drug_categories = [node_info.category]
                 # add the node to the knowledge graph
                 drug_name = node_info.name

diff --git a/code/ARAX/test/test_ARAX_expand.py b/code/ARAX/test/test_ARAX_expand.py
@@ -1302,7 +1302,7 @@ def test_xdtd_curie_not_in_db():
             "t_edge": {
                 "object": "disease",
                 "subject": "chemical",
-                "predicates": ["biolink:affects"],
+                "predicates": ["biolink:treats"],
                 "knowledge_type": "inferred"
             }
         }