Fix keys parameter in edges_iter

Fixes #813
This commit is contained in:
Bob Lantz
2018-07-11 23:36:11 +00:00
parent 966dd20b4b
commit 26b3959968
+3 -3
View File
@@ -56,13 +56,13 @@ class MultiGraph( object ):
return self.node.items() if data else self.node.keys() return self.node.items() if data else self.node.keys()
def edges_iter( self, data=False, keys=False ): def edges_iter( self, data=False, keys=False ):
"Iterator: return graph edges" "Iterator: return graph edges, optionally with data and keys"
for src, entry in self.edge.iteritems(): for src, entry in self.edge.iteritems():
for dst, keys in entry.iteritems(): for dst, entrykeys in entry.iteritems():
if src > dst: if src > dst:
# Skip duplicate edges # Skip duplicate edges
continue continue
for k, attrs in keys.iteritems(): for k, attrs in entrykeys.iteritems():
if data: if data:
if keys: if keys:
yield( src, dst, k, attrs ) yield( src, dst, k, attrs )