code: def setUp(self):
      super().setUp()
      self.lambda_ = 0.75
  
      self.q_tm1 = np.array(
          [
              [[1.1, 2.1], [-1.1, 1.1], [3.1, -3.1]],
              [[2.1, 3.1], [-1.1, 0.1], [-2.1, -1.1]],
          ],
          dtype=np.float32,
      )
      self.a_tm1 = np.array([[0, 1, 0], [1, 0, 0]], dtype=np.int32)
      self.discount_t = np.array(
          [[0.0, 0.89, 0.85], [0.88, 1.0, 0.83]], dtype=np.float32
      )
      self.r_t = np.array(
          [[-1.3, -1.3, 2.3], [1.3, 5.3, -3.3]], dtype=np.float32
      )
      self.q_t = np.array(
          [
              [[1.2, 2.2], [-1.2, 0.2], [2.2, -1.2]],
              [[4.2, 2.2], [1.2, 1.2], [-1.2, -2.2]],
          ],
          dtype=np.float32,
      )
  
      self.expected_td = np.array(
          [
              [[-2.4, 0.4280, 1.07], [0.6935, 3.478, -2.196]],
              [[-1.9329, 0.6643, -0.7854], [-0.20713, 2.1855, 0.27132]],
              [[-1.6179, 0.4633, -0.7576], [-1.1097, 1.6509, 0.3598]],
              [[-2.1785, 0.6562, -0.5938], [-0.0892, 2.6553, -0.1208]],
          ],
          dtype=np.float32,
      )
  
imports: import functools
  from absl.testing import absltest
  from absl.testing import parameterized
  import chex
  import jax
  import numpy
  from rlax._src import nonlinear_bellman
label_desc: Use image node_id 1 for calling the TransformedQLambdaTest obj's underlying member method code with example usage: obj.setUp() without return types