test CI

blackjax-devs · Jan 21, 2025 · 7b7588b · 7b7588b
1 parent a0812be
commit 7b7588b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/blackjax/adaptation/adjusted_mclmc_adaptation.py b/blackjax/adaptation/adjusted_mclmc_adaptation.py
@@ -328,8 +328,8 @@ def step(state, key):
             return next_state, next_state.position
 
         state, samples = jax.lax.scan(
-            f=step,
             init=state,
+            f=step,
             xs=adaptation_L_keys,
         )