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