def return_list(func): import inspect from functools import wraps assert inspect.isgeneratorfunction(func) @wraps(func) def wrapped(*args, **kargs): return list(func(*args, **kargs)) return wrapped