fix test because new jax 0.9.2 release [v2]#2156

juanitorduz merged 1 commit into

pyro-ppl:masterpyro-ppl/numpyro:masterfrom

juanitorduz:fix-error-handlingjuanitorduz/numpyro:fix-error-handlingCopy head branch name to clipboard