code: def input_generator(self):
      for _ in range(self.BATCH_SIZE):
          vals = np.random.randint(
              1, 100, size=np.random.randint(20), dtype=np.int32
          )
          yield {"inputs": vals}
  
file_path: tensor2tensor/tensor2tensor/utils/checkpoint_compatibility_test.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import numpy
  from six.moves import range
  from tensor2tensor import models
  from tensor2tensor import problems
  from tensor2tensor.utils import data_reader
  from tensor2tensor.utils import trainer_lib
  import tensorflow.compat.v1
label_desc: Use image node_id 3 for calling the CheckpointCompatibilityTest obj's underlying member method code with example usage: obj.input_generator() without return types