Fixed exception handling for --calc-L option

RUBi-ZA · Jun 4, 2017 · 5328f95 · 5328f95
1 parent 51c2485
commit 5328f95
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/calc_network.py b/calc_network.py
@@ -99,8 +99,11 @@ def calc_shortest_path(protein_graph, prefix, generate_plots=True):
 
     for i in range(num_nodes):
         for j in range(num_nodes):
-            dj_path_matrix[i,j] = path_dict[i][j]
-
+            try:
+                dj_path_matrix[i,j] = path_dict[i][j]
+            except KeyError, ke:
+                raise nx.exception.NetworkXNoPath("No link between %d and %d" % (i, j))
+
     np.savetxt("%s_L.dat" % prefix, dj_path_matrix)
 
     avg_L_per_node = np.sum(dj_path_matrix, axis=0)/(num_nodes - 1)