From 185a7a51b1a89a73e8f0f44a8c972cb9882b0732 Mon Sep 17 00:00:00 2001 From: Kevin Vizhalil Date: Wed, 16 Aug 2023 17:59:14 -0400 Subject: [PATCH] Add support graph only on creative edges #2022 --- code/ARAX/ARAXQuery/result_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/code/ARAX/ARAXQuery/result_transformer.py b/code/ARAX/ARAXQuery/result_transformer.py index 195ebd0e0..65bac6512 100644 --- a/code/ARAX/ARAXQuery/result_transformer.py +++ b/code/ARAX/ARAXQuery/result_transformer.py @@ -81,7 +81,7 @@ def transform(response: ARAXResponse): else: inferred_qedge_key = inferred_qedge_keys[0] inferred_edge_keys = {edge_binding.id for edge_binding in - result.analyses[0].edge_bindings[inferred_qedge_key]} + result.analyses[0].edge_bindings[inferred_qedge_key] if "creative_" in edge_binding.id} # Refer to the support graph from the proper edge(s) for inferred_edge_key in inferred_edge_keys: inferred_edge = message.knowledge_graph.edges[inferred_edge_key]