def test_diagonal_gaussian_prior_sample_from_p_z(): """ DiagonalGaussianPrior.sample_from_p_z works without crashing """ prior = DiagonalGaussianPrior() vae = DummyVAE() prior.set_vae(vae) prior.initialize_parameters(nhid=5) prior.sample_from_p_z(10)
def test_diagonal_gaussian_prior_sample_from_p_z(): """ DiagonalGaussianPrior.sample_from_p_z works without crashing """ prior = DiagonalGaussianPrior() vae = DummyVAE() prior.set_vae(vae) prior.initialize_parameters(nhid=5) prior.sample_from_p_z(10)