Make adapdict ready for KL-divergence regularization