code: def test_call(self):
      logits = layers.RestoreList()(
          inputs=(self.flattened_logits, self.mask)
      )
      self.assertAllEqual([2, 3], logits.get_shape().as_list())
      self.assertAllEqual(
          [
              [1, 0.5, tf.math.log(_EPSILON)],
              [0, tf.math.log(_EPSILON), tf.math.log(_EPSILON)],
          ],
          logits.numpy(),
      )
      logits = layers.RestoreList()(
          inputs=(self.flattened_logits_2d, self.mask)
      )
      self.assertAllEqual([2, 3], logits.get_shape().as_list())
      self.assertAllEqual(
          [
              [1, 0.5, tf.math.log(_EPSILON)],
              [0, tf.math.log(_EPSILON), tf.math.log(_EPSILON)],
          ],
          logits.numpy(),
      )
  
imports: import tensorflow
  from tensorflow_ranking.python.keras import layers
label_desc: Use image node_id 2 for calling the RestoreListTest obj's underlying member method code with example usage: obj.test_call() without return types