From a3e235f5f6cc2408f038c6ab858d1f49310d452c Mon Sep 17 00:00:00 2001
From: Reuben <reubenharry@users.noreply.github.com>
Date: Mon, 3 Jun 2024 01:31:10 -0400
Subject: [PATCH] FIX KWARG BUG (#686)

* FIX KWARG BUG

* FIX KWARG BUG
---
 blackjax/mcmc/mclmc.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/blackjax/mcmc/mclmc.py b/blackjax/mcmc/mclmc.py
index 27b5c2e9c..e7a69849b 100644
--- 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