update g reshape (#126)

QuantEcon · Sep 29, 2023 · 1924e23 · 1924e23
1 parent ccc296c
commit 1924e23
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/lectures/markov_asset.md b/lectures/markov_asset.md
@@ -977,6 +977,7 @@ def A(g, sv_model, shapes):
     P = np.reshape(P,            (I, 1, 1, I,  1,  1))
     Q = np.reshape(Q,            (1, J, 1, 1,  J,  1))
     R = np.reshape(R,            (1, 1, K, 1,  1,  K))
+    g = np.reshape(g,            (1, 1, 1, I,  J,  K))
     a = μ_d - γ * μ_c
     b = bar_σ**2 * (jnp.exp(2 * hd) + γ**2 * jnp.exp(2 * hc)) / 2
     κ = jnp.exp(a + (1 - γ) * z + b)