Add support graph only on creative edges #2022

RTXteam · Aug 16, 2023 · 185a7a5 · 185a7a5
1 parent 970ce9b
commit 185a7a5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]