code: def _test_dataset_ints(inp_lengths, tgt_lengths):
      """Create a test dataset of int64 tensors of given shapes."""
  
      def generator():
          for inp_len, tgt_len in zip(inp_lengths, tgt_lengths):
              inp = np.ones([inp_len], dtype=np.int64)
              tgt = np.ones([tgt_len], dtype=np.int64)
              yield {"inputs": inp, "targets": tgt}
  
      types = {"inputs": tf.int64, "targets": tf.int64}
      shapes = {
          "inputs": tf.TensorShape([None]),
          "targets": tf.TensorShape([None]),
      }
      return tf.data.Dataset.from_generator(
          generator, output_types=types, output_shapes=shapes
      )
  
imports: import collections
  import os
  from unittest import mock
  import gin
  import numpy
  from t5.data import assert_dataset
  from t5.data import preprocessors
  import tensorflow
  import tensorflow_datasets
  from trax.data import inputs
  from trax.data import tf_inputs
label_desc: Use image node_id 1 for calling a global function with example usage: _test_dataset_ints(inp_lengths, tgt_lengths) and returns: tf