diff --git a/BayesianKDEy/_bayeisan_kdey.py b/BayesianKDEy/_bayeisan_kdey.py index 197deaf..3c58900 100644 --- a/BayesianKDEy/_bayeisan_kdey.py +++ b/BayesianKDEy/_bayeisan_kdey.py @@ -291,7 +291,7 @@ class BayesianKDEy(AggregativeSoftQuantifier, KDEBase, WithConfidenceABC): if alpha is not None: alpha = jnp.asarray(alpha) numpyro.factor( - 'log_prior', dist.Dirichlet(alpha.log_prob(prev)) + 'log_prior', dist.Dirichlet(alpha).log_prob(prev) ) # if alpha is None, then this corresponds to a weak logistic-normal prior