code: def test_partition_axis_unspecified(self):
      def f(scope, xs):
          def g(scope, x):
              kernel_init = meta.with_partitioning(
                  nn.initializers.ones_init(), ("in", "out")
              )
              scope.param("kernel", kernel_init, (3, 2))
              return x
  
          with self.assertRaises(errors.PartitioningUnspecifiedError):
              lift.vmap(
                  g,
                  in_axes=0,
                  variable_axes={"params": 0},
                  split_rngs={"params": True},
                  metadata_params={},
              )(scope, xs)
  
      init(f)(random.key(0), jnp.zeros((8, 3)))
  
file_path: flax/tests/core/core_meta_test.py
imports: import jax
  from absl.testing import absltest
  from jax import numpy
  from jax import random, sharding
  from jax.experimental import mesh_utils
  from flax import errors
  from flax.core import init, lift, meta, nn
label_desc: Use image node_id 3 for calling the MetaTest obj's underlying member method code with example usage: obj.test_partition_axis_unspecified() and returns: x