From db60b84edefbb7b97bc7767b1c51a018f8559270 Mon Sep 17 00:00:00 2001 From: Ethan Dalool Date: Tue, 25 May 2021 01:45:34 -0700 Subject: [PATCH] Don't bother with 1-element list, just do it right away. --- voussoirkit/pipeable.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/voussoirkit/pipeable.py b/voussoirkit/pipeable.py index caf2a76..50222b1 100644 --- a/voussoirkit/pipeable.py +++ b/voussoirkit/pipeable.py @@ -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)