diff --git a/desc/magnetic_fields/_core.py b/desc/magnetic_fields/_core.py index 7cc1841d8..a598a1565 100644 --- a/desc/magnetic_fields/_core.py +++ b/desc/magnetic_fields/_core.py @@ -1592,8 +1592,10 @@ def default_terminating_event_fxn(state, **kwargs): # suppress warnings till its fixed upstream: # https://github.com/patrick-kidger/diffrax/issues/445 + # also ignore deprecation warning for now until we actually need to deal with it with warnings.catch_warnings(): warnings.filterwarnings("ignore", message="unhashable type") + warnings.filterwarnings("ignore", message="`diffrax.*discrete_terminating") x = jnp.vectorize(intfun, signature="(k)->(n,k)")(x0) x = jnp.where(jnp.isinf(x), jnp.nan, x)