Fix VAE tests failing when ran from the models/tests directory
Merge pull request #1242 from vdumoulin/fix_vae_tests

Fix VAE tests failing when ran from the models/tests directory (#1241)