Change reset_burn_in to actually work for sub-classes of Sampler