fix: do not set tfinal to inf

alexfikl · Oct 4, 2023 · 250202e · 250202e
1 parent b410658
commit 250202e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pyshocks/timestepping.py b/pyshocks/timestepping.py
@@ -106,7 +106,8 @@ def step(
     :returns: a :class:`StepCompleted` at the end of each taken time step.
     """
     if tfinal is None:
-        tfinal = jnp.inf
+        # NOTE: can't set this to jnp.inf because we have a debug check for it
+        tfinal = jnp.finfo(u0.dtype).max
 
     m = 0
     t = jnp.array(tstart, dtype=u0.dtype)