snapshot...

modules/deeplearning/cloud_fraction_fcn.py

@@ -820,7 +820,7 @@ def run_evaluate_static(in_file, out_file, ckpt_dir):
     var_names = ['cloud_fraction', 'temp_11_0um', 'refl_0_65um']
     dims = ['num_params', 'y', 'x']
 
-    da = xr.DataArray(np.column_stack([cld_frac_out, bt, refl_avg]), dims=dims)
+    da = xr.DataArray(np.stack([cld_frac_out, bt, refl_avg], axis=0), dims=dims)
     da.assign_coords({
         'lat': (['y', 'x'], lats),
         'lon': (['y', 'x'], lons)