-def force_generator(generator_function):
- @functools.wraps(generator_function)
- def forced_generator(*args, **kwargs):
- return list(generator_function(*args, **kwargs))
+def force_generator(to_type):
+ def decorator(generator_function):
+ @functools.wraps(generator_function)
+ def forced_generator(*args, **kwargs):
+ return to_type(generator_function(*args, **kwargs))