Don't bother with 1-element list, just do it right away.
This commit is contained in:
parent
4741946eda
commit
db60b84ede
1 changed files with 2 additions and 1 deletions
|
@ -151,7 +151,8 @@ def input_many(args, *input_args, **input_kwargs):
|
|||
This saves you from having to write the double for loop yourself.
|
||||
'''
|
||||
if isinstance(args, str):
|
||||
args = [args]
|
||||
yield from input(args, *input_args, **input_kwargs)
|
||||
return
|
||||
|
||||
for arg in args:
|
||||
yield from input(arg, *input_args, **input_kwargs)
|
||||
|
|
Loading…
Reference in a new issue