code: def test_nan_tracetime():
      @jax.jit
      def f(x):
          return eqx.error_if(x, True, "hi", on_error="nan")
  
      with pytest.warns(UserWarning):
          y = f(1.0)
      assert jnp.isnan(y)
  
file_path: equinox/tests/test_errors.py
imports: import equinox
  import equinox.internal
  import jax
  import jax.numpy
  import pytest
label_desc: Use image node_id 8 for calling a global function with example usage: test_nan_tracetime() and returns: eqx