Update interfaces.py

ciemss · Aug 27, 2024 · 6c0ca3c · 6c0ca3c
1 parent a8d2714
commit 6c0ca3c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyciemss/interfaces.py b/pyciemss/interfaces.py
@@ -537,7 +537,7 @@ def wrapped_model():
         for k, vals in samples.items():
             if "_state" in k:
                 # Note: qoi is assumed to be the last day of simulation
-                qoi_sample = vals[..., 0, :].detach().numpy()[:, -1]
+                qoi_sample = vals.detach().numpy()[:, -1]
                 sq_est = alpha_superquantile(qoi_sample, alpha=alpha)
                 risk_results.update({k: {"risk": [sq_est], "qoi": qoi_sample}})