fixed how callable list is read in 'ouu.py'

ciemss · Jul 16, 2024 · 646f1a3 · 646f1a3
1 parent d5ef867
commit 646f1a3
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/pyciemss/ouu/ouu.py b/pyciemss/ouu/ouu.py
@@ -89,9 +89,7 @@ def __init__(
         risk_bound: float = 0.0,
     ):
         self.model = model
-        self.interventions = (
-            [interventions] if not isinstance(interventions, list) else interventions
-        )
+        self.interventions = interventions
         self.qoi = qoi
         self.risk_measure = risk_measure
         self.num_samples = num_samples
@@ -139,7 +137,11 @@ def propagate_uncertainty(self, x):
                 intervention_list = [
                     deepcopy(self.fixed_static_parameter_interventions)
                 ]
-                intervention_list.extend(self.interventions(torch.from_numpy(x)))
+                intervention_list.extend(
+                    [self.interventions(torch.from_numpy(x))]
+                    if not isinstance(self.interventions(torch.from_numpy(x)), list)
+                    else self.interventions(torch.from_numpy(x))
+                )
                 static_parameter_interventions = combine_static_parameter_interventions(
                     intervention_list
                 )