Removed call to reconstruct_state in test_pretrain_SdA.py