code: def test_max_traces():
      @jax.jit
      @eqx.debug.assert_max_traces(max_traces=1)
      def f(x):
          return x + 1
  
      f(1)
      f(2)
  
      with pytest.raises(
          RuntimeError, match="can only be traced 1 times"
      ):
          f(3.0)
  
file_path: equinox/tests/test_debug.py
imports: import equinox
  import jax
  import jax.numpy
  import jax.tree_util
  import pytest
label_desc: Use image node_id 3 for calling a global function with example usage: test_max_traces() and returns: unknown