Update test_interfaces.py

ciemss · Nov 13, 2024 · eadbe3e · eadbe3e
1 parent d2634ed
commit eadbe3e
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/test_interfaces.py b/tests/test_interfaces.py
@@ -652,10 +652,22 @@ def __call__(self, x):
         opt_intervention = opt_intervention_temp
     if "fixed_static_state_interventions" not in optimize_kwargs:
         fixed_static_state_interventions = {}
+    else:
+        fixed_static_state_interventions = optimize_kwargs[
+            "fixed_static_state_interventions"
+        ]
     if "fixed_dynamic_parameter_interventions" not in optimize_kwargs:
         fixed_dynamic_parameter_interventions = {}
+    else:
+        fixed_dynamic_parameter_interventions = optimize_kwargs[
+            "fixed_dynamic_parameter_interventions"
+        ]
     if "fixed_dynamic_state_interventions" not in optimize_kwargs:
         fixed_dynamic_state_interventions = {}
+    else:
+        fixed_dynamic_state_interventions = optimize_kwargs[
+            "fixed_dynamic_state_interventions"
+        ]
 
     result_opt = sample(
         model_url,