add parameters to observables

ciemss · Jun 5, 2024 · db587da · db587da
1 parent 7967dfa
commit db587da
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/pyciemss/mira_integration/compiled_dynamics.py b/pyciemss/mira_integration/compiled_dynamics.py
@@ -158,8 +158,14 @@ def _eval_observables_mira(
 ) -> State[torch.Tensor]:
     if len(src.observables) == 0:
         return dict()
+
+    parameters = {
+        get_name(param_info): getattr(param_module, get_name(param_info))
+        for param_info in src.parameters.values()
+        if not param_info.placeholder
+    }
 
-    numeric_observables = param_module.numeric_observables_func(**X)
+    numeric_observables = param_module.numeric_observables_func(**X, **parameters)
 
     observables: State[torch.Tensor] = dict()
     for i, obs in enumerate(src.observables.values()):