334c0750a62000155366bc8257f950f420dec3f1,tensorflow_transform/beam/analysis_graph_builder.py,,_serialize_op_attr,#Any#,41

Before Change

          "Unable to serialize op attributes that contain a `list.func` field")
    if attr_value.HasField("func"):
      // TODO(b/138796127): Support tf.function fingerprint.
      result.append(uuid.uuid4().hex)
      attr_value.ClearField("func")
    result.append(attr_value.SerializeToString())
  return result

After Change

def _serialize_op_attr(op_attr):
  Deterministicly serializes tf.Operation attrs since it is a map.
  sorted_attributes = sorted(op_attr.items(), key=lambda kv: kv[0])
  if "f" in op_attr:
    // This is a tf.Function node, and it includes attributes that are
    // inconsistent across runs such as _gradient_op_type, config_proto, so we
    // only keep input and output types since other information will arrive from
    // the FuncGraph attributes.
    sorted_attributes = [
        kv for kv in sorted_attributes if kv[0] in ("Tin", "Tout")
    ]
  result = []
  for key, attr_value in sorted_attributes:
    result.append(key)
    attr_value = copy.deepcopy(attr_value)

Instances