manage levels for 'multiple included lists' (#46359)

* manage levels for 'multiple included lists'

fixes #46343
This commit is contained in:
Brian Coca 2018-10-10 12:34:58 -04:00 committed by GitHub
parent 878599fd9f
commit 80d977bac6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -440,8 +440,8 @@ def flatten(mylist, levels=None):
if levels is None:
ret.extend(flatten(element))
elif levels >= 1:
levels = int(levels) - 1
ret.extend(flatten(element, levels=levels))
# decrement as we go down the stack
ret.extend(flatten(element, levels=(int(levels) - 1)))
else:
ret.append(element)
else: