Update README.md (#719)

blackjax-devs · Aug 13, 2024 · 4a11236 · 4a11236
1 parent 834f55d
commit 4a11236
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -76,8 +76,8 @@ state = nuts.init(initial_position)
 # Iterate
 rng_key = jax.random.key(0)
 step = jax.jit(nuts.step)
-for step in range(100):
-    nuts_key = jax.random.fold_in(rng_key, step)
+for i in range(100):
+    nuts_key = jax.random.fold_in(rng_key, i)
     state, _ = step(nuts_key, state)
 ```