code: def setUp(self):
      super().setUp()
      self.flattened_logits = tf.constant(
          [1, 0.5, 2, 0, -1, 0], dtype=tf.float32
      )
      self.flattened_logits_2d = tf.constant(
          [[1], [0.5], [2], [0], [-1], [0]], dtype=tf.float32
      )
      self.mask = tf.constant(
          [[True, True, False], [True, False, False]], dtype=tf.bool
      )
  
imports: import tensorflow
  from tensorflow_ranking.python.keras import layers
label_desc: Use image node_id 1 for calling the RestoreListTest obj's underlying member method code with example usage: obj.setUp() without return types