Provide a fix for `sample_blackjax_nuts` failing with `chains=1` with prior parameters of different shapes#5969

twiecki merged 1 commit intopymc-devs:mainpymc-devs/pymc:mainfrom

bherwerth:fix-sampling-jax-5954bherwerth/pymc:fix-sampling-jax-5954Copy head branch name to clipboard