remove adaptation_window filter - blocked on jaxlib

pymc-devs · Jul 11, 2024 · f43f002 · f43f002
1 parent 2588b2d
commit f43f002
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/pymc/sampling/jax.py b/pymc/sampling/jax.py
@@ -27,7 +27,6 @@
 import pytensor.tensor as pt
 
 from arviz.data.base import make_attrs
-from blackjax.adaptation.base import get_filter_adapt_info_fn
 from jax.lax import scan
 from pytensor.compile import SharedVariable, Supervisor, mode
 from pytensor.graph.basic import graph_inputs
@@ -393,7 +392,6 @@ def _sample_blackjax_nuts(
         tune=tune,
         draws=draws,
         target_accept=target_accept,
-        adaptation_info_fn=get_filter_adapt_info_fn(),
         **nuts_kwargs,
     )