Skip to content

Commit 3711f5b

Browse files
Update lectures/mccall_model.md
Co-authored-by: Humphrey Yang <39026988+HumphreyYang@users.noreply.github.com>
1 parent 01c5e80 commit 3711f5b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

lectures/mccall_model.md

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -823,7 +823,10 @@ def compute_R_element(c, β):
823823
824824
# Create meshgrid and vectorize computation
825825
c_grid, β_grid = jnp.meshgrid(c_vals, β_vals, indexing='ij')
826-
compute_R_vectorized = jax.vmap(jax.vmap(compute_R_element, in_axes=(None, 0)), in_axes=(0, None))
826+
compute_R_vectorized = jax.vmap(
827+
jax.vmap(compute_R_element,
828+
in_axes=(None, 0)),
829+
in_axes=(0, None))
827830
R = compute_R_vectorized(c_vals, β_vals)
828831
```
829832

0 commit comments

Comments
 (0)