249cd01eccb5c787779c69292d6262b514c0df8b,tensorflow_transform/saved/saved_transform_io_test.py,SavedTransformIOTest,test_table_roundtrip,#SavedTransformIOTest#,158

Before Change

      with tf.compat.v1.Session().as_default() as session:
        input_string = tf.compat.v1.placeholder(tf.string)
        // Map string through a table, in this case based on a constant tensor.
        table = lookup_ops.index_table_from_tensor(
            tf.constant(["cat", "dog", "giraffe"]))
        output = table.lookup(input_string)
        inputs = {"input": input_string}
        outputs = {"output": output}
        saved_transform_io.write_saved_transform_from_session(

After Change

        input_string = tf.compat.v1.placeholder(tf.string)
        // Map string through a table, in this case based on a constant tensor.
        table_keys = ["cat", "dog", "giraffe"]
        initializer = tf.lookup.KeyValueTensorInitializer(
            keys=table_keys,
            values=tf.cast(tf.range(len(table_keys)), tf.int64),
            key_dtype=tf.string,
            value_dtype=tf.int64)
        table = tf.lookup.StaticHashTable(initializer, default_value=-1)

        output = table.lookup(input_string)
        inputs = {"input": input_string}
        outputs = {"output": output}

Instances