Fix multivariate reconstruction

fzhu2e · Nov 11, 2024 · 2cc8d57 · 2cc8d57
1 parent 6f2b5dc
commit 2cc8d57
Show file tree

Hide file tree

Showing 2 changed files with 85 additions and 98 deletions.
diff --git a/cfr/da/enkf.py b/cfr/da/enkf.py
@@ -119,8 +119,7 @@ def gen_Xb(self):
         # vn_1st = list(self.prior.keys())[0]
         Xb_var_irow = {}  # index of rows in Xb to store the specific var
         loc = 0
-        i = 0
-        for vn in self.recon_vars:
+        for i, vn in enumerate(self.recon_vars):
             _, nlat, nlon = np.shape(self.prior[vn].da.values)
             lats, lons = self.prior[vn].da.lat.values, self.prior[vn].da.lon.values
             lon2d, lat2d = np.meshgrid(lons, lats)

diff --git a/docsrc/notebooks/psm-linear.ipynb b/docsrc/notebooks/psm-linear.ipynb