diff --git a/pyciemss/interfaces.py b/pyciemss/interfaces.py index 81f87645..9f523c15 100644 --- a/pyciemss/interfaces.py +++ b/pyciemss/interfaces.py @@ -532,7 +532,7 @@ def wrapped_model(): risk_results = {} for k, vals in samples.items(): - if "_state" in k: + if "_state" in k or "_observable" in k: # qoi is assumed to be the last day of simulation qoi_sample = vals.detach().numpy()[:, -1] sq_est = alpha_superquantile(qoi_sample, alpha=alpha)