merge main

blackjax-devs · Aug 7, 2024 · 3f9947c · 3f9947c
1 parent 3c7daf2
commit 3f9947c
Showing 1 changed file with 0 additions and 6 deletions.
diff --git a/blackjax/util.py b/blackjax/util.py
@@ -200,14 +200,8 @@ def one_step(state, xs):
         state, info = inference_algorithm.step(rng_key, state)
         return state, transform(state, info)
 
-    xs = (jnp.arange(num_steps), keys)
     if progress_bar:
         one_step = progress_bar_scan(num_steps)(one_step)
-        (((_, average), final_state), _), history = lax.scan(
-            one_step,
-            (((0, expectation(transform(initial_state))), initial_state), -1),
-            xs,
-        )
 
     xs = jnp.arange(num_steps), keys
     final_state, history = lax.scan(one_step, initial_state, xs)