Skip to content

Commit 8ce1c85

Browse files
author
Ben Cipollini
committed
BF: avoid sorting with tuple
1 parent 2d572d5 commit 8ce1c85

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

nipype/pipeline/engine.py

Lines changed: 4 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -778,13 +778,10 @@ def _set_needed_outputs(self, graph):
778778
node.needed_outputs = []
779779
for edge in graph.out_edges_iter(node):
780780
data = graph.get_edge_data(*edge)
781-
for sourceinfo, _ in sorted(data['connect']):
782-
if isinstance(sourceinfo, tuple):
783-
input_name = sourceinfo[0]
784-
else:
785-
input_name = sourceinfo
786-
if input_name not in node.needed_outputs:
787-
node.needed_outputs += [input_name]
781+
sourceinfo = [v1[0] if isinstance(v1, tuple) else v1
782+
for v1, v2 in data['connect']]
783+
node.needed_outputs += [v for v in sourceinfo
784+
if v not in node.needed_outputs]
788785
if node.needed_outputs:
789786
node.needed_outputs = sorted(node.needed_outputs)
790787

0 commit comments

Comments
 (0)