fix bug

MESMER-group · Apr 3, 2022 · 7d286f8 · 7d286f8
1 parent 439bac3
commit 7d286f8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/mesmer/calibrate_mesmer/train_gv.py b/mesmer/calibrate_mesmer/train_gv.py
@@ -208,7 +208,7 @@ def train_gv_AR(params_gv, gv, max_lag, sel_crit):
 
         params_scen.append(params)
 
-    params_scen = xr.concat(res, dim="scen")
+    params_scen = xr.concat(params_scen, dim="scen")
     params_scen = params_scen.mean("scen")
 
     # TODO: remove np.float64(...) (only here so the tests pass)

diff --git a/mesmer/calibrate_mesmer/train_lv.py b/mesmer/calibrate_mesmer/train_lv.py
@@ -240,7 +240,7 @@ def train_lv_AR1_sci(params_lv, targs, y, wgt_scen_eq, aux, cfg):
 
             params_scen.append(params)
 
-        params_scen = xr.concat(res, dim="scen")
+        params_scen = xr.concat(params_scen, dim="scen")
         params_scen = params_scen.mean("scen")
 
         params_lv["AR1_int"][targ_name] = params_scen.intercept.values