We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8f8719f commit cfdc30dCopy full SHA for cfdc30d
test/test_handlers.py
@@ -657,6 +657,7 @@ def model():
657
x = numpyro.sample("x", dist.Exponential(1))
658
with handlers.collapse():
659
with numpyro.plate("d", d):
660
+ # TODO: verify that to_event works here
661
beta0 = numpyro.sample("beta0", dist.Normal(0, 1).expand([S]).to_event(1))
662
# TODO: address beta0 is a str, which cannot do infer_param_domain
663
beta = numpyro.sample("beta", dist.MultivariateNormal(beta0, jnp.eye(S)))
0 commit comments