# we should not disconnect from upstream if
# there are still elements depending on us.
assert len(self.downstream_elements) == 0, "there are still downstream elements left"
- if self.source:
+
+ # Sources don't have a source themselves. don't do anything here.
+ if self.source is not None:
self.source.disconnectDownstream(self)
def disconnectDownstream(self, downstream):