f73cdc286a3e125cf7019336621cb10370ebfd52,tensorflow_transform/analyzer_nodes.py,_VocabularyAccumulatorCoder,encode_cache,#_VocabularyAccumulatorCoder#Any#,393

Before Change

  def encode_cache(self, accumulator):
    // Need to wrap in np.array and call tolist to make it JSON serializable.
    word, count = accumulator
    if self._input_dtype == tf.string:
      word = tf.compat.as_text(word)
    accumulator = (word, count)

After Change

    // If the value is a _WeightedMeanAndVarAccumulator, cast each field to a
    // list for serialization.
    try:
      value = value._replace(
          count=value.count.tolist(),
          mean=value.mean.tolist(),
          variance=value.variance.tolist(),
          weight=value.weight.tolist())
    except AttributeError:
      pass
    accumulator = (token, value)
    return tf.compat.as_bytes(
        json.dumps(np.array(accumulator, dtype=object).tolist()))

  def decode_cache(self, encoded_accumulator):

Instances