5b2fa49f9ee24bd079ab0c8d5373a3bfcfcc60b4,tensorflow_transform/impl_helper_test.py,ImplHelperTest,testInferFeatureSchema,#ImplHelperTest#,54

Before Change

                                    sch.LogicalShape([])),
            sch.FixedColumnRepresentation()),
        "b": sch.ColumnSchema(
            sch.LogicalColumnSchema(sch.dtype_to_domain(tf.string),
                                    sch.LogicalShape([sch.Axis(2),
                                                      sch.Axis(3)])),
            sch.FixedColumnRepresentation()),
        "c": sch.ColumnSchema(
            sch.LogicalColumnSchema(sch.dtype_to_domain(tf.int64),

After Change

    expected_schema = sch.Schema(column_schemas={
        "a": sch.ColumnSchema(tf.float32, [],
                              sch.FixedColumnRepresentation()),
        "b": sch.ColumnSchema(tf.string, [2, 3],
                              sch.FixedColumnRepresentation()),
        "c": sch.ColumnSchema(tf.int64, None,
                              sch.FixedColumnRepresentation())

Instances