fix

AMICI-dev · Nov 19, 2024 · c242b15 · c242b15
1 parent a585414
commit c242b15
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tests/test_jax.py b/python/tests/test_jax.py
@@ -162,7 +162,7 @@ def check_fields_jax(
     )
     fun = beartype(jax_model.simulate_condition)
 
-    for output in ["nllh", "x0", "x", "y", "res"]:
+    for output in ["llh", "x0", "x", "y", "res"]:
         oargs = (*args[:-2], diffrax.DirectAdjoint(), 2**8, output)
         if sensi_order == amici.SensitivityOrder.none:
             r_jax[output] = fun(p, *oargs)[0]