Ignore diffrax deprecation warning for now

PlasmaControl · Aug 27, 2024 · 2dab0a3 · 2dab0a3
1 parent 59f9175
commit 2dab0a3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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)