Make simulate_petab return the generated/used ExpDatas (#1933)

Quite helpful for debugging.
AMICI-dev · Jan 16, 2023 · 7f1eee3 · 7f1eee3
1 parent 11a1f27
commit 7f1eee3
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/sdist/amici/petab_objective.py b/python/sdist/amici/petab_objective.py
@@ -37,6 +37,7 @@
 RES = 'res'
 SRES = 'sres'
 RDATAS = 'rdatas'
+EDATAS = 'edatas'
 
 
 @log_execution_time('Simulating PEtab model', logger)
@@ -92,6 +93,7 @@ def simulate_petab(
 
         * cost function value (``LLH``),
         * list of :class:`amici.amici.ReturnData` (``RDATAS``),
+        * list of :class:`amici.amici.ExpData` (``EDATAS``),
 
         corresponding to the different simulation conditions.
         For ordering of simulation conditions, see
@@ -163,7 +165,8 @@ def simulate_petab(
 
     return {
         LLH: llh,
-        RDATAS: rdatas
+        RDATAS: rdatas,
+        EDATAS: edatas,
     }