code: def test_compile_time_eval():
      class A(eqxi.Enumeration):
          a = "hi"
          b = "bye"
  
      @jax.jit
      def f(pred):
          x = A.where(True, A.a, A.b)
          assert not x.is_traced()
          y = x == A.a
          assert y
          z = A.where(pred, A.a, A.b)
          assert z.is_traced()
  
      f(True)
  
file_path: equinox/tests/test_enum.py
imports: import equinox
  import equinox.internal
  import jax
  import jax.numpy
  import jax.tree_util
  import pytest
  from .helpers import tree_allclose
label_desc: Use image node_id 10 for calling a global function with example usage: test_compile_time_eval() without return types