Merge pull request #1500 from qingqing01/topology_fix

Fix __bfs_travel__ in topology.py
avx_docs
hedaoyuan 8 years ago committed by GitHub
commit cda45793b1

@ -21,7 +21,21 @@ import layer as v2_layer
__all__ = ['Topology']
def __flatten__(lis):
"""
Given a list, possibly nested to any level, return it flattened.
"""
new_lis = []
for item in lis:
if isinstance(item, collections.Sequence):
new_lis.extend(__flatten__(item))
else:
new_lis.append(item)
return new_lis
def __bfs_travel__(callback, *layers):
layers = __flatten__(layers)
for each_layer in layers:
__break__ = callback(each_layer)
if __break__:

Loading…
Cancel
Save