Suppress FutureWarning in trajectory calculation to avoid cluttering output (see see patrick-kidger/diffrax#445)

Author: EstevaoMGomes

essos/dynamics.py

@@ -237,6 +237,8 @@ def update_state(state, _):
                 _, trajectory = lax.scan(update_state, initial_condition, jnp.arange(len(self.times)-1))
                 trajectory = jnp.vstack([initial_condition, trajectory])
             else:
+                import warnings
+                warnings.simplefilter("ignore", category=FutureWarning) # see https://github.com/patrick-kidger/diffrax/issues/445 for explanation
                 trajectory = diffeqsolve(
                     self.ODE_term,
                     t0=0.0,