Skip to content

Commit 62daf27

Browse files
committed
update
1 parent a225ed9 commit 62daf27

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

lectures/career.md

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -410,8 +410,8 @@ $(\theta_i, \epsilon_j)$ = either 1, 2 or 3; meaning 'stay put',
410410

411411
```{code-cell} ipython3
412412
model = create_career_worker_problem()
413-
F = jnp.cumsum(model.F_probs)
414-
G = jnp.cumsum(model.G_probs)
413+
F = np.array(jnp.cumsum(model.F_probs))
414+
G = np.array(jnp.cumsum(model.G_probs))
415415
v_star = solve_model(model)
416416
greedy_star = get_greedy_policy(model, v_star)
417417
@@ -486,8 +486,8 @@ The median for the original parameterization can be computed as follows
486486

487487
```{code-cell} ipython3
488488
model = create_career_worker_problem()
489-
F = jnp.cumsum(model.F_probs)
490-
G = jnp.cumsum(model.G_probs)
489+
F = np.array(jnp.cumsum(model.F_probs))
490+
G = np.array(jnp.cumsum(model.G_probs))
491491
v_star = solve_model(model)
492492
greedy_star = get_greedy_policy(model, v_star)
493493

0 commit comments

Comments
 (0)