code: def test_ellipsis_not_last_error(self):
      m = nnx.Dict(
          a=nnx.Sequence([nnx.Param(1), nnx.Param(2), nnx.Variable(3)]),
          b=nnx.Dict(c=nnx.Param(1), d=nnx.BatchStat(2)),
      )
  
      with pytest.raises(
          ValueError,
          match="Ellipsis `...` can only be used as the last filter,",
      ):
          m.split(..., nnx.Param)
  
file_path: flax/flax/experimental/nnx/tests/test_partitioning.py
imports: import jax
  import pytest
  from flax.experimental import nnx
label_desc: Use image node_id 5 for calling the TestPartitioning obj's underlying member method code with example usage: obj.test_ellipsis_not_last_error() without return types