test_allclose jax.errors.TracerArrayConversionError: The numpy.ndarray conversion method __array__() was called on traced array with shape float32[] The error occurred while tracing the function <lambda> at C:\hostedtoolcache\windows\Python\3.12.7\x64\Lib\site-packages\quax\_core.py:546 for jit. This concrete value was not available in Python because it depends on the value of the argument x[1]. See https://jax.readthedocs.io/en/latest/errors.html#jax.errors.TracerArrayConversionError -------------------- For simplicity, JAX has removed its internal frames from the traceback of the following exception. Set JAX_TRACEBACK_FILTERING=off to include these.