code: def test_unbox(self):
      xs = {
          "kernel": meta.Partitioned(jnp.zeros((3, 2)), ("in", "out")),
          "complex": meta.Partitioned(
              {"K": jnp.zeros((3, 2)), "b": jnp.zeros((3,))}, ("data",)
          ),
      }
      unboxed = meta.unbox(xs)
      unboxed_shapes = jax.tree_map(jnp.shape, unboxed)
      self.assertEqual(
          unboxed_shapes,
          {
              "kernel": (3, 2),
              "complex": {
                  "K": (3, 2),
                  "b": (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 4 for calling the MetaTest obj's underlying member method code with example usage: obj.test_unbox() without return types