Update test_jax.py

AMICI-dev · Dec 18, 2024 · 353d6bb · 353d6bb
1 parent e40de89
commit 353d6bb
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/python/tests/test_jax.py b/python/tests/test_jax.py
@@ -199,6 +199,7 @@ def check_fields_jax(
         "solver": diffrax.Kvaerno5(),
         "controller": diffrax.PIDController(atol=ATOL_SIM, rtol=RTOL_SIM),
         "adjoint": diffrax.RecursiveCheckpointAdjoint(),
+        "steady_state_event": diffrax.steady_state_event(),
         "max_steps": 2**8,  # max_steps
     }
     fun = beartype(jax_model.simulate_condition)