fix get I in non base spend

PSLmodels · Oct 2, 2024 · 0d72a5b · 0d72a5b
1 parent 85b9e62
commit 0d72a5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ogcore/TPI.py b/ogcore/TPI.py
@@ -687,7 +687,7 @@ def run_TPI(p, client=None):
         D = np.zeros(p.T + p.S)
         D_d = np.zeros(p.T + p.S)
         D_f = np.zeros(p.T + p.S)
-        I_g = np.ones_like(Y) * ss_vars["I_g_ss"]
+        I_g = fiscal.get_I_g(Y[: p.T], None, p, "TPI")
     else:
         if p.baseline_spending:
             # Will set to TRbaseline here, but will be updated in TPI loop