Skip to content

Commit f641c86

Browse files
committed
extract some functionality into private funcs and confirm it works for end-to-end large ensemble run
1 parent 56b70ca commit f641c86

2 files changed

Lines changed: 528 additions & 355 deletions

File tree

pyCIAM/io.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -342,7 +342,9 @@ def _load_lslr_for_ciam(
342342
slr_out.indexes["scen_mc"], pd.MultiIndex
343343
):
344344
slr_out = slr_out.reset_index("scen_mc").assign_coords(
345-
scen_mc=slr_out.scenario.values + "_" + slr_out.sample.values.astype(str)
345+
scen_mc=slr_out.scenario.values
346+
+ "_"
347+
+ slr_out[mc_dim if quantiles is None else "quantile"].values.astype(str)
346348
)
347349
return slr_out
348350

@@ -704,8 +706,6 @@ def load_ciam_inputs(
704706
],
705707
dim="scen_mc",
706708
)
707-
if scen_mc_filter is None:
708-
slr = slr.unstack("scen_mc")
709709

710710
slr = slr.sel({k: v for k, v in selectors.items() if k in slr.dims})
711711

0 commit comments

Comments
 (0)