Skip to content

GitLab

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

Fix test

parent fef8ef1e
Show whitespace changes
Inline Side-by-side
test/test_kl.py
View file @ fdcda8cc
...@@ -133,8 +133,7 @@ def test_ParametricVI(mirror_samples, fc): ...@@ -133,8 +133,7 @@ def test_ParametricVI(mirror_samples, fc):
assert_allclose(true_val.val, kl.value, rtol=0.1) assert_allclose(true_val.val, kl.value, rtol=0.1)
samples = model.KL.samples() samples = model.KL.samples()
mini = ift.SteepestDescent(ift.GradientNormController(iteration_limit=3)) model.minimize(ift.ADVIOptimizer(ift.GradientNormController(iteration_limit=3)))
model.minimize(mini)
samples1 = model.KL.samples() samples1 = model.KL.samples()
for aa, bb in zip(samples, samples1): for aa, bb in zip(samples, samples1):
ift.extra.assert_allclose(aa, bb) 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