I am trying to produce a mixture model of StudentT distribution, but I am getting an error message that is very obscure to me about a NameError: unbound axis name: _provenance.
The very simplified model below still generates the error.
def model_at_given_mag(x=None, y=None, peak=None):
# Adjust peak location
loc = numpyro.sample('peak', dist.Normal(peak, 0.1))
# stable component
stable_logshape = numpyro.sample('s_logshape', dist.Uniform(1, 3))
stable_logscale = numpyro.sample('s_logscale', dist.Uniform(-3, 0))
stable_shape = 10 ** stable_logshape
stable_scale = 10 ** stable_logscale
stable = dist.StudentT(stable_shape, loc=loc, scale=stable_scale)
with numpyro.plate('obs', y.shape[0]):
# yobs represents the median at the pdf x location
obs = numpyro.sample("yobs", stable, obs=y)
Can anyone spot the issue?
It is very hard to find where the issue is from the traceback and error messages.