code: def test_arange_nonconcrete():
      """JAX cannot JIT-compile `jax.numpy.arange` when arguments are not concrete values."""
  
      a = scalar("a")
      a.tag.test_value = 10
      out = ptb.arange(a)
  
      with pytest.raises(NotImplementedError):
          fgraph = FunctionGraph([a], [out])
          compare_jax_and_py(
              fgraph, [get_test_value(i) for i in fgraph.inputs]
          )
  
imports: import numpy
  import pytest
  from pytensor.compile import get_mode
  import jax.errors
  import pytensor
  import pytensor.tensor.basic
  from pytensor.configdefaults import config
  from pytensor.graph.fg import FunctionGraph
  from pytensor.graph.op import get_test_value
  from pytensor.tensor.type import iscalar, matrix, scalar, vector
  from tests.link.jax.test_basic import compare_jax_and_py
  from tests.tensor.test_basic import TestAlloc
label_desc: Use image node_id 6 for calling a global function with example usage: test_arange_nonconcrete() without return types