Skip to content
Snippets Groups Projects
Commit 05e025b2 authored by Hartung, Michael's avatar Hartung, Michael
Browse files

targeet nodes without seeds in multisteiner

Former-commit-id: 2c41558490cef50c478d279fc88d07920432c088 [formerly 0d8e50317bf17a363aa12e2ae6f74cc3a54d5cdb]
Former-commit-id: 5114e5afe26fc8900512f08dd5ec84a553e540a0
parent a99c6a31
No related branches found
No related tags found
No related merge requests found
......@@ -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
})
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment