bug fix

modules/util/util.py

@@ -542,8 +542,8 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
     n_x = int(xlen/w_x)
     n_y = int(ylen/w_y)
 
-    ll = [j_0 + j*w_y for j in range(n_y-1)]
-    cc = [i_0 + i*w_x for i in range(n_x-1)]
+    # ll = [j_0 + j*w_y for j in range(n_y-1)]
+    # cc = [i_0 + i*w_x for i in range(n_x-1)]
 
     for ds_name in name_list:
         for j in range(n_y-1):
@@ -552,6 +552,16 @@ def make_for_full_domain_predict(h5f, name_list=None, satellite='GOES16', domain
                 i_ul = i * w_x
                 grd_dct_n[ds_name].append(grd_dct[ds_name][j_ul:j_ul+w_y, i_ul:i_ul+w_x])
 
+    # TODO: replace this with numpy meshgrid and flatten
+    ll = []
+    cc = []
+    for j in range(n_y-1):
+        j_y = j_0 + j*w_y
+        for i in range(n_x-1):
+            i_x = i_0 + i*w_x
+            ll.append(j_y)
+            cc.append(i_x)
+
     grd_dct = {name: None for name in name_list}
     for didx, ds_name in enumerate(name_list):
         grd_dct[ds_name] = np.stack(grd_dct_n[ds_name])