5b2fa49f9ee24bd079ab0c8d5373a3bfcfcc60b4,tensorflow_transform/tf_metadata/dataset_schema.py,,_from_parse_feature,#Any#,381

Before Change

  if isinstance(parse_feature, tf.SparseFeature):
    sparse_shape = LogicalShape(
        axes=[Axis(parse_feature.size)])
    logical = LogicalColumnSchema(
        domain=_dtype_to_domain(parse_feature.dtype),
        shape=sparse_shape)
    index_field = SparseIndexField(name=parse_feature.index_key,
                                   is_sorted=parse_feature.already_sorted)
    representation = SparseColumnRepresentation(
        value_field_name=parse_feature.value_key,
        index_fields=[index_field])
    return ColumnSchema(logical, representation)

  raise ValueError("Cannot interpret feature spec: {}".format(parse_feature))

After Change

    representation = SparseColumnRepresentation(
        value_field_name=parse_feature.value_key,
        index_fields=[index_field])
    return ColumnSchema(parse_feature.dtype, [parse_feature.size],
                        representation)

  raise ValueError("Cannot interpret feature spec: {}".format(parse_feature))

Instances