fix output dict construction

AMICI-dev · Dec 5, 2024 · 9a25420 · 9a25420
1 parent 9d82a6c
commit 9a25420
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/sdist/amici/jax/petab.py b/python/sdist/amici/jax/petab.py
@@ -421,4 +421,7 @@ def run_simulations(
         )
         for sc in simulation_conditions
     }
-    return sum(llh for llh, _ in results.values()), results | preeqs
+    return sum(llh for llh, _ in results.values()), {
+        sc: res[1] | preeqs[sc[1]][1] if len(sc) > 1 else res[1]
+        for sc, res in results.items()
+    }