FIX KWARG BUG (#686)

* FIX KWARG BUG * FIX KWARG BUG
blackjax-devs · Jun 3, 2024 · a3e235f · a3e235f
1 parent 3fbdac6
commit a3e235f
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/blackjax/mcmc/mclmc.py b/blackjax/mcmc/mclmc.py
@@ -80,7 +80,9 @@ def build_kernel(logdensity_fn, sqrt_diag_cov, integrator):
 
     """
 
-    step = with_isokinetic_maruyama(integrator(logdensity_fn, sqrt_diag_cov))
+    step = with_isokinetic_maruyama(
+        integrator(logdensity_fn=logdensity_fn, sqrt_diag_cov=sqrt_diag_cov)
+    )
 
     def kernel(
         rng_key: PRNGKey, state: IntegratorState, L: float, step_size: float