remove debug print

sammccallum · Nov 26, 2024 · 8a7448e · 8a7448e
1 parent 0391bc1
commit 8a7448e
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/diffrax/_adjoint.py b/diffrax/_adjoint.py
@@ -1031,7 +1031,6 @@ def cond_fun(state):
         grad_terms,
     )
 
-    jax.debug.print("{}", t1_only)
     state = eqxi.while_loop(cond_fun, grad_step, state, kind="lax")
     _, _, _, grad_ys, grad_z0, grad_args, grad_terms = state
     if t1_only: