File tree Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Expand file tree Collapse file tree 1 file changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -410,13 +410,13 @@ def _generate_dependency_list(self, graph):
410
410
""" Generates a dependency list for a list of graphs.
411
411
"""
412
412
self .procs , _ = topological_sort (graph )
413
- nodes = graph .nodes ()
414
- indices = [nodes .index (proc ) for proc in self .procs ]
415
413
try :
416
- self .depidx = nx .to_scipy_sparse_matrix (graph , format = 'lil' )
414
+ self .depidx = nx .to_scipy_sparse_matrix (graph ,
415
+ nodelist = self .procs ,
416
+ format = 'lil' )
417
417
except :
418
- self .depidx = nx .to_scipy_sparse_matrix (graph )
419
- self . depidx = self . depidx [:, indices ][ indices , :]
418
+ self .depidx = nx .to_scipy_sparse_matrix (graph ,
419
+ nodelist = self . procs )
420
420
self .refidx = deepcopy (self .depidx )
421
421
self .refidx .astype = np .int
422
422
self .proc_done = np .zeros (len (self .procs ), dtype = bool )
You can’t perform that action at this time.
0 commit comments