code: def test_no_arrays():
      @eqx.filter_vmap(
          out_axes=_zero_if_inexact_array_else_none, axis_size=5
      )
      def f(x):
          return x
  
      assert tree_allclose(f(1), 1)
  
file_path: equinox/tests/test_vmap.py
imports: from typing import Union
  import equinox
  import jax
  import jax.numpy
  import jax.random
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 5 for calling a global function with example usage: test_no_arrays() and returns: x