ensure G and TR are T+S long

jdebacker · Aug 3, 2024 · e1ab01e · e1ab01e
1 parent 19c90fc
commit e1ab01e
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/ogcore/TPI.py b/ogcore/TPI.py
@@ -665,9 +665,8 @@ def run_TPI(p, client=None):
         D_f = np.zeros(p.T + p.S)
     else:
         if p.baseline_spending:
-            TR = TRbaseline
-            G = Gbaseline
-            G[p.T :] = ss_vars["Gss"]
+            TR = np.concatenate((TRbaseline[:p.T], np.ones(p.S) * ss_vars["TR_ss"]))
+            G = np.concatenate((Gbaseline[:p.T], np.ones(p.S) * ss_vars["Gss"]))
         else:
             TR = p.alpha_T * Y
             G = np.ones(p.T + p.S) * ss_vars["Gss"]