Add KLAT to creative expand treats edges #2412

RTXteam · Nov 15, 2024 · 4c261d8 · 4c261d8
1 parent 07e6fe2
commit 4c261d8
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/code/ARAX/ARAXQuery/ARAX_expander.py b/code/ARAX/ARAXQuery/ARAX_expander.py
@@ -542,7 +542,13 @@ def apply(self, response, input_parameters, mode: str = "ARAX"):
                             # Create a lumped edge to represent all of these edges
                             lumped_edge = Edge(subject=subj_key, object=obj_key, predicate="biolink:treats",
                                                sources=[RetrievalSource(resource_id="infores:arax",
-                                                                        resource_role="primary_knowledge_source")])
+                                                                        resource_role="primary_knowledge_source")],
+                                               attributes=[Attribute(attribute_type_id="biolink:agent_type",
+                                                                     value="computational_model",
+                                                                     attribute_source="infores:arax"),
+                                                           Attribute(attribute_type_id="biolink:knowledge_level",
+                                                                     value="not_provided",
+                                                                     attribute_source="infores:arax")])
                             lumped_edge_key = f"creative_expand_treats_edge:{subj_key}--treats--{obj_key}--infores:arax"
                             overarching_kg.edges_by_qg_id[qedge_key][lumped_edge_key] = lumped_edge