cb7c20f6ad714f0b016a874b2b06a9fbed4e0a59,tensorflow_transform/test_case.py,TransformTestCase,AssertVocabularyContents,#TransformTestCase#Any#Any#,290

Before Change

      raise

  def AssertVocabularyContents(self, vocab_file_path, file_contents):
    with tf.io.gfile.GFile(vocab_file_path, "rb") as f:
      file_lines = f.readlines()

      // Store frequency case.

After Change

          tf_utils.read_tfrecord_vocabulary_dataset(
              vocab_file_path).as_numpy_iterator())
    else:
      with tf.io.gfile.GFile(vocab_file_path, "rb") as f:
        file_lines = f.read().splitlines()

    // Store frequency case.
    if isinstance(file_contents[0], tuple):
      word_and_frequency_list = []
      for content in file_lines:
        frequency, word = content.split(b" ", 1)

Instances