format

PSLmodels · Aug 5, 2024 · 64d690c · 64d690c
1 parent e1ab01e
commit 64d690c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ogcore/TPI.py b/ogcore/TPI.py
@@ -665,8 +665,12 @@ def run_TPI(p, client=None):
         D_f = np.zeros(p.T + p.S)
     else:
         if p.baseline_spending:
-            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"]))
+            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"]