Allow custom edges in algorithms

Former-commit-id: c1d7e66b

tasks/trust_rank.py

@@ -213,7 +213,7 @@ def trust_rank(task_hook: TaskHook):
     if custom_edges:
       edges = task_hook.parameters.get("input_network")['edges']
       g = add_edges(g, edges)
-    
+      
     task_hook.set_progress(1 / 4.0, "Computing edge weights.")
     weights = edge_weights(g, hub_penalty, inverse=True)
     

tasks/util/custom_edges.py

@@ -11,8 +11,10 @@ def add_edges(g, edge_list):
     mapping = make_node_id_map(g)
     edge_id_list = []
     for edge in edge_list:
-        a = mapping[edge['from']]
-        b = mapping[edge['to']]
-        edge_id_list.append((a, b))
-    g.add_edge_list(edge_list)
+        a = mapping[edge['from']] if edge['from'] in mapping else False
+        b = mapping[edge['to']] if edge['to'] in mapping else False
+        if a and b:
+            edge_id_list.append((a, b, 'protein-protein'))
+    e_type = g.edge_properties["type"]
+    g.add_edge_list(edge_id_list, eprops=[e_type])
     return g
\ No newline at end of file