code: def test_wrong():
      sentinel1 = object()
      sentinel2 = object()
  
      def f(x, y):
          assert y is sentinel1
          return x + 1, sentinel1
  
      out_struct = (jax.ShapeDtypeStruct((), jnp.float32), sentinel2)
  
      with pytest.raises(RuntimeError):
          eqx.filter_pure_callback(
              f,
              jnp.array(1.0),
              sentinel1,
              result_shape_dtypes=out_struct,
          )
  
      @eqx.filter_jit
      def g(x):
          return eqx.filter_pure_callback(
              f, x, sentinel1, result_shape_dtypes=out_struct
          )
  
      with pytest.raises(RuntimeError):
          g(jnp.array(2.0))
  
file_path: equinox/tests/test_callback.py
imports: import equinox
  import jax
  import jax.numpy
  import numpy
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 2 for calling a global function with example usage: test_wrong() and returns: unknown,  sentinel1, eqx