Merge pull request #2 from jdebacker/pr-871

black formatting
PSLmodels · Oct 4, 2023 · bd11970 · bd11970
2 parents 263872b + 261e6db
commit bd11970
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ogcore/TPI.py b/ogcore/TPI.py
@@ -262,7 +262,7 @@ def twist_doughnut(
     n_s = n_guess
     chi_n_s = p.chi_n[-length:]
     e_s = p.e[-length:, j]
-    rho_s = np.diag(p.rho[t : t + p.S, :], max(p.S - (s + 1), 0))
+    rho_s = np.diag(p.rho[t : t + p.S, :], max(p.S - length, 0))
 
     error1 = household.FOC_savings(
         r_s,