bugfix for custom_edges input bug

drugst-one · Oct 9, 2024 · e0acf39 · e0acf39
1 parent 3fe6770
commit e0acf39
Show file tree

Hide file tree

Showing 7 changed files with 20 additions and 2 deletions.
diff --git a/drugstone/management/commands/import_from_nedrex.py b/drugstone/management/commands/import_from_nedrex.py
@@ -252,7 +252,7 @@ def add_dpi(edge):
                 if update and e.__hash__() in existing:
                     if existing[e.__hash__()] != e:
                         delete.add(existing[e.__hash__()])
-                        del existing[e.__hash__()]
+                        del existing[e.__hash__()]  
                     else:
                         return
                 if not update or e.__hash__() not in existing:

diff --git a/tasks/betweenness_centrality.py b/tasks/betweenness_centrality.py
@@ -169,11 +169,14 @@ def betweenness_centrality(task_hook: TaskHook):
 
     search_target = task_hook.parameters.get("target", "drug-target")
 
-    filterPaths = task_hook.parameters.get("filter_paths", True)
+    filterPaths = task_hook.parameters.get("filter_paths", True)  
 
     id_space = task_hook.parameters["config"].get("identifier","symbol")
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges =    no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)
 

diff --git a/tasks/closeness_centrality.py b/tasks/closeness_centrality.py
@@ -175,6 +175,9 @@ def closeness_centrality(task_hook: TaskHook):
     id_space = task_hook.parameters["config"].get("identifier", "symbol")
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges =    no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)
 

diff --git a/tasks/degree_centrality.py b/tasks/degree_centrality.py
@@ -150,6 +150,9 @@ def degree_centrality(task_hook: TaskHook):
     filterPaths = task_hook.parameters.get("filter_paths", True)
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges =    no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)
 

diff --git a/tasks/multi_steiner.py b/tasks/multi_steiner.py
@@ -101,6 +101,9 @@ def multi_steiner(task_hook: TaskHook):
     node_name_attribute = "internal_id" # nodes in the input network which is created from RepoTrialDB have primaryDomainId as name attribute
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges =    no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)
 

diff --git a/tasks/network_proximity.py b/tasks/network_proximity.py
@@ -81,6 +81,9 @@ def network_proximity(task_hook: TaskHook):
     filter_paths = task_hook.parameters.get("filter_paths", True)
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges =    no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)
 

diff --git a/tasks/trust_rank.py b/tasks/trust_rank.py
@@ -198,6 +198,9 @@ def trust_rank(task_hook: TaskHook):
     filter_paths = task_hook.parameters.get("filter_paths", True)
 
     custom_edges = task_hook.parameters.get("custom_edges", False)
+    if custom_edges:
+        if not isinstance(custom_edges, list):
+            custom_edges = False
 
     no_default_edges = task_hook.parameters.get("exclude_drugstone_ppi_edges", False)