Skip to content

Commit d3a847c

Browse files
committed
minor variable name update
1 parent 7b5bb5d commit d3a847c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

lectures/mccall_correlated.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -335,7 +335,7 @@ def compute_unemployment_duration(model,
335335
return jnp.interp(z, z_grid, f_star)
336336
337337
@jax.jit
338-
def draw_tau(key, t_max=10_000):
338+
def draw_τ(key, t_max=10_000):
339339
def cond_fun(state):
340340
z, t, unemployed, key = state
341341
return jnp.logical_and(unemployed, t < t_max)
@@ -373,7 +373,7 @@ def compute_unemployment_duration(model,
373373
keys = jr.split(key, num_reps)
374374
375375
# Vectorize over simulations
376-
τ_vals = jax.vmap(draw_tau)(keys)
376+
τ_vals = jax.vmap(draw_τ)(keys)
377377
378378
return jnp.mean(τ_vals)
379379
```

0 commit comments

Comments
 (0)