Modify test to check ClippedAdam's learning rate decay argument (#9)

Author: neerajprad

pyroapi/tests/test_svi.py

@@ -53,8 +53,11 @@ def model(data=None):
 
 
 @pytest.mark.parametrize("jit", [False, True], ids=["py", "jit"])
-@pytest.mark.parametrize("optim_name", ["Adam", "ClippedAdam"])
-def test_optimizer(backend, optim_name, jit):
+@pytest.mark.parametrize("optim_name, optim_kwargs", [
+    ("Adam", {"lr": 1e-6}),
+    ("ClippedAdam", {"lr": 1e-6, "lrd": 0.999}),
+])
+def test_optimizer(backend, optim_name, optim_kwargs, jit):
 
     def model(data):
         p = pyro.param("p", ops.tensor(0.5))
@@ -67,7 +70,7 @@ def guide(data):
     pyro.get_param_store().clear()
     Elbo = infer.JitTrace_ELBO if jit else infer.Trace_ELBO
     elbo = Elbo(ignore_jit_warnings=True)
-    optimizer = getattr(optim, optim_name)({"lr": 1e-6})
+    optimizer = getattr(optim, optim_name)(optim_kwargs.copy())
     inference = infer.SVI(model, guide, optimizer, elbo)
     for i in range(2):
         inference.step(data)