Update qoi.py

ciemss · Aug 10, 2024 · 233ae7c · 233ae7c
1 parent 4f54dd3
commit 233ae7c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyciemss/ouu/qoi.py b/pyciemss/ouu/qoi.py
@@ -14,7 +14,7 @@ def obs_nday_average_qoi(
     Note: last ndays timepoints is assumed to represent last n-days of simulation.
     """
     dataQoI = samples[contexts[0]].detach().numpy()
-
+    dataQoI = np.atleast_2d(np.squeeze(dataQoI))
     return np.mean(dataQoI[:, -ndays:], axis=1)
 
 
@@ -25,5 +25,5 @@ def obs_max_qoi(samples: Dict[str, torch.Tensor], contexts: List) -> np.ndarray:
     samples[VARIABLE] is expected to have dimension (nreplicates, ntimepoints)
     """
     dataQoI = samples[contexts[0]].detach().numpy()
-
+    dataQoI = np.atleast_2d(np.squeeze(dataQoI))
     return np.max(dataQoI, axis=1)