ec9dcf7491c38f3531bfa718b496861ab19bb048,tensorflow_transform/analyzer_nodes.py,_VocabularyAccumulatorCoder,encode_cache,#_VocabularyAccumulatorCoder#Any#,430

Before Change

      token = tf.compat.as_text(token)
    // 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()))

After Change

    // list for serialization.
    if isinstance(value, tuple):
      value = [
          a.tolist()
          for a in (value.count, value.mean, value.variance, value.weight)
      ]
    value = tf.compat.as_bytes(json.dumps(value))
    len_token, len_value = len(token), len(value)

Instances