Make generator_args with a list comprehension