diff --git a/examples/ar2.py b/examples/ar2.py index 744d8925e..b6f0e737a 100644 --- a/examples/ar2.py +++ b/examples/ar2.py @@ -64,6 +64,8 @@ def transition(carry, _): with numpyro.handlers.condition(data={"y": y[2:]}): _, mu = scan(transition, init, timesteps) + numpyro.deterministic("mu", mu) + def ar2_for_loop(y): alpha_1 = numpyro.sample("alpha_1", dist.Normal(0, 1))