code: def test_disable_jit_odeint_with_vmap(self):
      # https://github.com/google/jax/issues/2598
      with jax.disable_jit():
          t = jnp.array([0.0, 1.0])
          x0_eval = jnp.zeros((5, 2))
          f = lambda x0: odeint(lambda x, _t: x, x0, t)
          jax.vmap(f)(x0_eval)
  
imports: from functools import partial
  from absl.testing import absltest
  import numpy
  import jax
  from jax._src import test_util
  import jax.numpy
  from jax.experimental.ode import odeint
  from jax.tree_util import tree_map
  import scipy.integrate
  from jax import config
label_desc: Use image node_id 9 for calling the ODETest obj's underlying member method code with example usage: obj.test_disable_jit_odeint_with_vmap() without return types