From 05e025b2dbd681c4d55da0651b8e63240b955c14 Mon Sep 17 00:00:00 2001
From: "Hartung, Michael" <michael.hartung@uni-hamburg.de>
Date: Mon, 13 Feb 2023 14:51:52 +0100
Subject: [PATCH] targeet nodes without seeds in multisteiner

Former-commit-id: 2c41558490cef50c478d279fc88d07920432c088 [formerly 0d8e50317bf17a363aa12e2ae6f74cc3a54d5cdb]
Former-commit-id: 5114e5afe26fc8900512f08dd5ec84a553e540a0
---
 tasks/multi_steiner.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tasks/multi_steiner.py b/tasks/multi_steiner.py
index 1567d39..c6a8055 100755
--- a/tasks/multi_steiner.py
+++ b/tasks/multi_steiner.py
@@ -180,6 +180,7 @@ def multi_steiner(task_hook: TaskHook):
                 returned_edges.append((node, int(neighbor)))
 
     accepted_nodes = [g.vertex_properties[node_name_attribute][node] for node in returned_nodes]
+    accepted_nodes_without_seeds = [g.vertex_properties[node_name_attribute][node] for node in returned_nodes if node not in seed_ids]
     subgraph = {"nodes": accepted_nodes,
                 "edges": [{"from": g.vertex_properties[node_name_attribute][source], "to": g.vertex_properties[node_name_attribute][target]} for
                           source, target in returned_edges]}
@@ -188,7 +189,7 @@ def multi_steiner(task_hook: TaskHook):
     task_hook.set_results({
         "network": subgraph,
         "node_attributes": {"node_types": node_types, "is_seed": is_seed},
-        "target_nodes": accepted_nodes,
+        "target_nodes": accepted_nodes_without_seeds,
         'gene_interaction_dataset': ppi_dataset,
         'drug_interaction_dataset': pdi_dataset
     })
-- 
GitLab