Wrap open_dataarray and open_mfdataset in with...

Author: measrainsey

message_ix_buildings/chilled/core/climate.py

@@ -156,31 +156,35 @@ def map_calculated_variables(args):
 
         log.info("Reading: " + filepath)
         if config.rcp == "rcp26":
-            dst = xr.open_mfdataset(
+            with xr.open_mfdataset(
                 filepath,
                 chunks={"lon": get_paths(config, "chunk_size")},
                 concat_dim="time",
                 use_cftime=True,
-            )  # Setting for RCP2.6
+            ) as ras:
+                dst = ras.load()
         else:
-            dst = xr.open_mfdataset(
+            with xr.open_mfdataset(
                 filepath,
                 chunks={"lon": get_paths(config, "chunk_size")},
-            )  # , concat_dim='time' )  # Setting for RCP6.0
+            ) as ras:
+                dst = ras.load()
 
         dst_crop = dst.sel(time=slice(years_clim[0], years_clim[1]))
         t_out_ave = dst_crop[config.davar].astype("float32") - 273.16
         t_out_ave = t_out_ave.transpose("lat", "lon", "time")
         t_oa_gbm = t_out_ave.groupby("time.month")
 
-        i_sol_v = xr.open_dataarray(
+        with xr.open_dataarray(
             os.path.join(dle_path, "EWEMBI_vert_irrad_1980-2009_avg.nc")
-        )  # Values  in daily Wh/m2
+        ) as ras:
+            i_sol_v = ras.load()  # Values in daily Wh/m2
 
         # Horizontal irradiation
-        i_sol_h = xr.open_dataarray(
+        with xr.open_dataarray(
             os.path.join(dle_path, "EWEMBI_horiz_irrad_1980-2009_avg.nc")
-        )  # Values in daily Wh/m2
+        ) as ras:
+            i_sol_h = ras.load()  # Values in daily Wh/m2
 
         if config.arch_setting == "regional":
             xr.open_dataset(
@@ -844,14 +848,15 @@ def aggregate_urban_rural_files(config: "Config"):
             for parset in par_var.itertuples():
                 for var in varlist:
                     log.info("." + var)
-                    mds = xr.open_mfdataset(
+                    with xr.open_mfdataset(
                         os.path.join(
                             output_path_vdd,
                             suff + "_" + str(parset.Index) + "_" + var + "_*.nc",
                         ),
                         concat_dim="urt",
                         combine="nested",
-                    )  # +urt[0]
+                    ) as ras:
+                        mds = ras.load()
                     # mds['arch'] = al #arcs#[:4]
                     mds["urt"] = config.urts  # arcs#[:4]