code: def test_pytree_in():
      @eqx.filter_jit
      @eqx.internal.noinline
      def fn(f, x):
          return f(x[0][0])
  
      o1 = fn(lambda x: x + 1, [(1,)])
      o2 = fn(lambda x: x + 1, ([jnp.array(1)],))
      assert tree_allclose(o1, 2)
      assert tree_allclose(o2, jnp.array(2))
  
file_path: equinox/tests/test_noinline.py
imports: import equinox
  import jax
  import jax.numpy
  import jax.random
  from .helpers import tree_allclose
label_desc: Use image node_id 7 for calling a global function with example usage: test_pytree_in() and returns: f