code: def _assert_no_unvmap(jaxpr: jax.core.Jaxpr):
      for eqn in jaxpr.eqns:
          assert eqn.primitive not in (
              eqxi.unvmap_any_p,
              eqxi.unvmap_all_p,
          )
      for subjaxpr in jax.core.subjaxprs(jaxpr):
          _assert_no_unvmap(subjaxpr)
  
file_path: equinox/tests/test_finalise_jaxpr.py
imports: from typing import cast
  import equinox.internal
  import jax
  import jax.core
  import jax.lax
  import jax.numpy
  from .helpers import tree_allclose
label_desc: Use image node_id 8 for calling a global function with example usage: _assert_no_unvmap(jaxpr) without return types