You'll need to set the ``buffer_size`` kwarg here instead, since it can change for the last iteration.