From 7fc85293b3e48cc4f7c662c9675cc634a0391226 Mon Sep 17 00:00:00 2001
From: amykglen <glena@oregonstate.edu>
Date: Tue, 19 Nov 2024 15:25:22 -0800
Subject: [PATCH] Refine KLAT for Expand's creative edges #2412

---
 code/ARAX/ARAXQuery/ARAX_expander.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/code/ARAX/ARAXQuery/ARAX_expander.py b/code/ARAX/ARAXQuery/ARAX_expander.py
index bb4d40f52..d509f0f32 100644
--- a/code/ARAX/ARAXQuery/ARAX_expander.py
+++ b/code/ARAX/ARAXQuery/ARAX_expander.py
@@ -544,10 +544,10 @@ def apply(self, response, input_parameters, mode: str = "ARAX"):
                                                sources=[RetrievalSource(resource_id="infores:arax",
                                                                         resource_role="primary_knowledge_source")],
                                                attributes=[Attribute(attribute_type_id="biolink:agent_type",
-                                                                     value="computational_model",
+                                                                     value="automated_agent",
                                                                      attribute_source="infores:arax"),
                                                            Attribute(attribute_type_id="biolink:knowledge_level",
-                                                                     value="not_provided",
+                                                                     value="prediction",
                                                                      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