Update ExampleJaxPEtab.ipynb

AMICI-dev · Dec 18, 2024 · 5cfe866 · 5cfe866
1 parent 353d6bb
commit 5cfe866
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/examples/example_jax_petab/ExampleJaxPEtab.ipynb b/python/examples/example_jax_petab/ExampleJaxPEtab.ipynb
@@ -378,6 +378,7 @@
     "        iy_trafos=jnp.array(iy_trafos),\n",
     "        solver=diffrax.Kvaerno5(),\n",
     "        controller=diffrax.PIDController(atol=1e-8, rtol=1e-8),\n",
+    "        steady_state_event=diffrax.steady_state_event(),\n",
     "        max_steps=2**10,\n",
     "        adjoint=diffrax.DirectAdjoint(),\n",
     "        ret=ReturnValue.y,  # Return observables\n",