Skip to content

Commit 195447e

Browse files
committed
change file loading behavior in benchmarking
1 parent cf9aeca commit 195447e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

batchglm/benchmark/nb_glm/base.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -214,9 +214,9 @@ def load_benchmark_dataset(root_dir: str, config_file="config.yml") -> Tuple[Sim
214214
logger.info("loading step-wise netcdf4 files...")
215215
ncdf_data = xr.open_mfdataset(
216216
os.path.join(root_dir, cfg["working_dir"], "estimation-*.h5"),
217-
engine="netcdf4",
217+
engine="h5netcdf",
218218
concat_dim="step",
219-
autoclose=True,
219+
# autoclose=True,
220220
parallel=True,
221221
)
222222
ncdf_data = ncdf_data.sortby("global_step")

0 commit comments

Comments
 (0)