zip_safe=False (@eigenstate)