Also ensure sortedness of incoming edge map

This commit is contained in:
Davide Depau 2018-01-26 16:05:56 +01:00
parent b7455d3261
commit b4e8c38b68
No known key found for this signature in database
GPG Key ID: C7D999B6A55EFE86

View File

@ -1,5 +1,7 @@
from __future__ import unicode_literals
from collections import OrderedDict
from .dag import KwargReprNode
from ._utils import escape_chars, get_hash_int
from builtins import object
@ -46,7 +48,7 @@ def get_stream_map(stream_spec):
elif isinstance(stream_spec, Stream):
stream_map = {None: stream_spec}
elif isinstance(stream_spec, (list, tuple)):
stream_map = dict(enumerate(stream_spec))
stream_map = OrderedDict(enumerate(stream_spec))
elif isinstance(stream_spec, dict):
stream_map = stream_spec
return stream_map
@ -84,7 +86,7 @@ class Node(KwargReprNode):
@classmethod
def __get_incoming_edge_map(cls, stream_map):
incoming_edge_map = {}
incoming_edge_map = OrderedDict()
for downstream_label, upstream in list(stream_map.items()):
incoming_edge_map[downstream_label] = (upstream.node, upstream.label)
return incoming_edge_map