I think that you can use chain.from_iterable to avoid the * trick.