Commit fdcda8cc authored by Philipp Arras's avatar Philipp Arras
Browse files

Fix test

parent fef8ef1e
Pipeline #103379 failed with stages
in 7 minutes and 1 second
......@@ -133,8 +133,7 @@ def test_ParametricVI(mirror_samples, fc):
assert_allclose(true_val.val, kl.value, rtol=0.1)
samples = model.KL.samples()
mini = ift.SteepestDescent(ift.GradientNormController(iteration_limit=3))
model.minimize(mini)
model.minimize(ift.ADVIOptimizer(ift.GradientNormController(iteration_limit=3)))
samples1 = model.KL.samples()
for aa, bb in zip(samples, samples1):
ift.extra.assert_allclose(aa, bb)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment