code: def test_jax_split_not_supported(self):
      a = matrix("a", shape=(6, None))
  
      a_splits = ptb.split(
          a, splits_size=[2, a.shape[1] - 2], n_splits=2, axis=1
      )
      with pytest.warns(
          UserWarning,
          match="Split node does not have constant split positions.",
      ):
          fn = pytensor.function([a], a_splits, mode="JAX")
      # It raises an informative ConcretizationTypeError, but there's an AttributeError that surpasses it
      with pytest.raises(AttributeError):
          fn(np.zeros((6, 4), dtype=pytensor.config.floatX))
  
      split_axis = iscalar("split_axis")
      a_splits = ptb.split(
          a, splits_size=[2, 4], n_splits=2, axis=split_axis
      )
      with pytest.warns(
          UserWarning, match="Split node does not have constant axis."
      ):
          fn = pytensor.function([a, split_axis], a_splits, mode="JAX")
      # Same as above, an AttributeError surpasses the `TracerIntegerConversionError`
      # Both errors are included for backwards compatibility
      with pytest.raises(
          (AttributeError, jax.errors.TracerIntegerConversionError)
      ):
          fn(np.zeros((6, 6), dtype=pytensor.config.floatX), 0)
  
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 3 for calling the TestJaxSplit obj's underlying member method code with example usage: obj.test_jax_split_not_supported() without return types