voussoir/voussoirkit
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Don't bother with 1-element list, just do it right away.

Browse source
This commit is contained in:
voussoir 2021-05-25 01:45:34 -07:00
parent 4741946eda
commit db60b84ede
No known key found for this signature in database
GPG key ID: 5F7554F8C26DACCB
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
voussoirkit/pipeable.py
View file

@ -151,7 +151,8 @@ def input_many(args, *input_args, **input_kwargs):
This saves you from having to write the double for loop yourself. This saves you from having to write the double for loop yourself.
''' '''
if isinstance(args, str): if isinstance(args, str):
args = [args] yield from input(args, *input_args, **input_kwargs)
return
for arg in args: for arg in args:
yield from input(arg, *input_args, **input_kwargs) yield from input(arg, *input_args, **input_kwargs)