Skip to content

Commit b30a084

Browse files
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

File tree

conda-envs/environment-alternative-backends.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,11 @@ dependencies:
3939
- types-cachetools
4040
# blackjax now does not pull in fastprogress by default
4141
- fastprogress>=1.0.0
42-
- blackjax>=1.5
42+
# blackjax 1.6 tightened kwarg checking on the NUTS/HMC kernel builder, which
43+
# exposes a latent bug in pymc.sampling.jax._blackjax_inference_loop: it
44+
# forwards progress_bar into blackjax.window_adaptation()'s **adaptation_kwargs
45+
# before popping it back out. Pin until that call is fixed to pop it first.
46+
- blackjax>=1.5,<1.6
4347
- pip:
4448
- numdifftools>=0.9.40
4549
- mcbackend>=0.4.0

0 commit comments

Comments
 (0)