From 353d6bbfb10ca208f26b5a8c8cbdb91f41dafa61 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20Fr=C3=B6hlich?= <fabian@schaluck.com>
Date: Wed, 18 Dec 2024 22:35:33 +0000
Subject: [PATCH] Update test_jax.py

---
 python/tests/test_jax.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/tests/test_jax.py b/python/tests/test_jax.py
index f8261fb15d..78fa026cfc 100644
--- 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)