improve type error message for 'fn' in sample function#2175

Qazalbash wants to merge 1 commit intopyro-ppl:masterpyro-ppl/numpyro:masterfrom

Qazalbash:modify-type-errorQazalbash/numpyro:modify-type-errorCopy head branch name to clipboard