Fixed the map_labels in the utils, fixing the unit tests

2 jobs for autoencoder_pretrain in 7 minutes and 59 seconds (queued for 1 second)