fix anneal for new rng handling
fix tpe for new rng handling