Fix bug for good and exclude test_vae.py from non-model tests