From fdcda8cc7dfc67bce36a9aae4cf5c2774d19c4b9 Mon Sep 17 00:00:00 2001
From: Philipp Arras <parras@mpa-garching.mpg.de>
Date: Fri, 11 Jun 2021 10:29:42 +0200
Subject: [PATCH] Fix test
---
test/test_kl.py | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/test/test_kl.py b/test/test_kl.py
index ad5d813f5..410ff3a9d 100644
--- 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)
--
GitLab