Skip to content

Commit cda4579

Browse files
authored
Merge pull request #1500 from qingqing01/topology_fix
Fix __bfs_travel__ in topology.py
2 parents be22171 + d447f06 commit cda4579

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

python/paddle/v2/topology.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,21 @@
2121
__all__ = ['Topology']
2222

2323

24+
def __flatten__(lis):
25+
"""
26+
Given a list, possibly nested to any level, return it flattened.
27+
"""
28+
new_lis = []
29+
for item in lis:
30+
if isinstance(item, collections.Sequence):
31+
new_lis.extend(__flatten__(item))
32+
else:
33+
new_lis.append(item)
34+
return new_lis
35+
36+
2437
def __bfs_travel__(callback, *layers):
38+
layers = __flatten__(layers)
2539
for each_layer in layers:
2640
__break__ = callback(each_layer)
2741
if __break__:

0 commit comments

Comments
 (0)