Fix test

fdcda8cc · Philipp Arras · fef8ef1e · fdcda8cc
Commit fdcda8cc authored Jun 11, 2021 by Philipp Arras
--- a/test/test_kl.py
+++ b/test/test_kl.py
@@ -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)