code: def setUp(self):
      super().setUp()
      self._prev_check_jax_usage = config.check_jax_usage(True)
  
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 1 for calling the LiftTest obj's underlying member method code with example usage: obj.setUp() without return types