code: def test_lift_raises_with_state(self):
      f = transform.transform_with_state(
          lambda: base.get_state("w", [], init=jnp.zeros)
      )
      lifted = lift.lift(f.init)  # pytype: disable=wrong-arg-types
      with self.assertRaisesRegex(
          base.NonEmptyStateError, "use.*lift_with_state"
      ):
          lifted(None)
  
imports: from absl.testing import absltest
  from absl.testing import parameterized
  from haiku._src import base
  from haiku._src import config
  from haiku._src import lift
  from haiku._src import module
  from haiku._src import multi_transform
  from haiku._src import stateful
  from haiku._src import test_utils
  from haiku._src import transform
  import jax
  import jax.numpy
  import numpy
label_desc: Use image node_id 10 for calling the LiftTest obj's underlying member method code with example usage: obj.test_lift_raises_with_state() without return types