code: def test_lift_module_called_repeatedly(self):
      def f():
          mod = transform_flax.lift(Outer(), name="foo")
          mod()
  
      f = transform.transform_with_state(f)
      rng = jax.random.PRNGKey(42)
      params, state = f.init(rng)
      self.assertEqual(params, {"foo/Parent_0/Child_0": {"w": 0}})
      self.assertEqual(
          state,
          {
              "foo/flim_state/Parent_0/Child_0": {"s": 1},
              "foo/flam_state/Parent_0/Child_0": {"s": 2},
          },
      )
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  import flax.errors
  import flax.linen
  from haiku._src import base
  from haiku._src import module
  from haiku._src import transform
  from haiku._src.flax import transform_flax
  import jax
  import jax.numpy
label_desc: Use image node_id 5 for calling the TransformFlaxTest obj's underlying member method code with example usage: obj.test_lift_module_called_repeatedly() without return types