3f5be904be6a354a5c3828982ec0d56c96fa835a,tensorflow_transform/tf_utils_test.py,VocabTFUtilsTest,test_read_tfrecord_vocabulary_dataset,#VocabTFUtilsTest#,1253

Before Change

    self._write_tfrecords(vocab_file, contents)
    self.AssertVocabularyContents(vocab_file, contents)

    ds = tf_utils.read_tfrecord_vocabulary_dataset(vocab_file)
    self.assertAllEqual(np.array(contents), list(ds.as_numpy_iterator()))

  @test_case.named_parameters([
      dict(

After Change

    self._write_tfrecords(vocab_file, contents)
    self.AssertVocabularyContents(vocab_file, contents)

    ds = tf.data.TFRecordDataset(vocab_file, compression_type="GZIP")
    self.assertAllEqual(np.array(contents), list(ds.as_numpy_iterator()))

  @test_case.named_parameters([
      dict(

Instances