Commit b30a084
committed
Pin blackjax<1.6 in alternative-backends conda env
blackjax 1.6 tightened kwarg validation on its NUTS/HMC kernel, which
surfaces a latent bug in pymc.sampling.jax._blackjax_inference_loop:
progress_bar is forwarded into blackjax.window_adaptation()'s
**adaptation_kwargs before being popped back out a few lines later,
so it leaks into the kernel builder. blackjax 1.5 tolerated the stray
kwarg; 1.6 raises TypeError. Pin until the wrapper pops progress_bar
before calling window_adaptation.1 parent 66c1c9e commit b30a084
1 file changed
Lines changed: 5 additions & 1 deletion
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
39 | 39 | | |
40 | 40 | | |
41 | 41 | | |
42 | | - | |
| 42 | + | |
| 43 | + | |
| 44 | + | |
| 45 | + | |
| 46 | + | |
43 | 47 | | |
44 | 48 | | |
45 | 49 | | |
| |||
0 commit comments