code: def test_call_by_scatter(self):
      logits = layers.RestoreList(by_scatter=True)(
          inputs=(self.flattened_logits, self.mask)
      )
      self.assertAllEqual([2, 3], logits.get_shape().as_list())
      self.assertAllClose(
          [
              [1.5, 0.5, tf.math.log(_EPSILON)],
              [
                  -1.0 / 3.0,
                  tf.math.log(_EPSILON),
                  tf.math.log(_EPSILON),
              ],
          ],
          logits.numpy(),
      )
      logits = layers.RestoreList(by_scatter=True)(
          inputs=(self.flattened_logits_2d, self.mask)
      )
      self.assertAllEqual([2, 3], logits.get_shape().as_list())
      self.assertAllClose(
          [
              [1.5, 0.5, tf.math.log(_EPSILON)],
              [
                  -1.0 / 3.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 3 for calling the RestoreListTest obj's underlying member method code with example usage: obj.test_call_by_scatter() without return types