assign value to TR_ss

PSLmodels · Sep 16, 2024 · 2028762 · 2028762
1 parent 6e665fc
commit 2028762
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/ogcore/SS.py b/ogcore/SS.py
@@ -1427,6 +1427,7 @@ def run_SS(p, client=None):
             p_m_ss = sol.x[3 : 3 + p.M]
             Yss = sol.x[3 + p.M]
             BQss = sol.x[3 + p.M + 1 : -1]
+            TR_ss = sol.x[-1]
         else:
             ss_params_reform = (
                 b_guess,

diff --git a/tests/test_SS.py b/tests/test_SS.py
@@ -660,7 +660,11 @@ def test_inner_loop_extra(baseline, param_updates, filename, dask_client):
     BQ = np.ones(p.J) * 0.00019646295986015257
     p_m = np.array([1.0])
     r_p = 0.04260341179572245
-    outer_loop_vars = (bssmat, nssmat, r_p, r, w, p_m, Y, BQ, TR, factor)
+    if p.baseline_spending:
+        Ig_baseline = 0.0
+    else:
+        Ig_baseline = None
+    outer_loop_vars = (bssmat, nssmat, r_p, r, w, p_m, Y, BQ, TR, Ig_baseline, factor)
     test_tuple = SS.inner_loop(outer_loop_vars, p, dask_client)
     expected_tuple = utils.safe_read_pickle(
         os.path.join(CUR_PATH, "test_io_data", filename)