Complain if an elipsis is in the chain

This commit is contained in:
Romain Dorgueil
2018-08-01 11:06:53 +01:00
parent 89dda0dca6
commit a4f07953a8

View File

@ -29,12 +29,13 @@ class GraphCursor:
def __rshift__(self, other): def __rshift__(self, other):
""" Self >> Other """ """ Self >> Other """
if other == ...: nodes = other.nodes if hasattr(other, "nodes") else [other]
if ... in nodes:
raise NotImplementedError( raise NotImplementedError(
"Expected something looking like a node, but got an Ellipsis (...). Did you forget to complete the graph?" "Expected something looking like a node, but got an Ellipsis (...). Did you forget to complete the graph?"
) )
nodes = other.nodes if hasattr(other, "nodes") else [other]
if len(nodes): if len(nodes):
chain = self.graph.add_chain(*nodes, _input=self.last) chain = self.graph.add_chain(*nodes, _input=self.last)