Fix bug in examples.mixture.main.compress_annealing