30def1a72c54b5635a2ed59bf03ecc10bef1a845,tensorflow_transform/impl_helper_test.py,ImplHelperTest,testCreatePhasesWithUnwrappedTable,#ImplHelperTest#,395

Before Change

      integerized = table.lookup(inputs["x"])
      return {"integerized": integerized}

    input_schema = sch.Schema({
        "x": sch.ColumnSchema(tf.string, [], sch.FixedColumnRepresentation())
    })
    _, _ = impl_helper.run_preprocessing_fn(
        preprocessing_fn, input_schema)
    with self.assertRaisesRegexp(ValueError, "Found table initializers"):
      _ = impl_helper.create_phases()

  def testCreatePhasesWithLoop(self):

After Change

  def testCreatePhasesWithUnwrappedTable(self):
    // Create a graph with a table that is not wrapped in `apply_function`.
    string_placeholder = tf.placeholder(tf.string, shape=(None,))
    table = lookup.index_table_from_tensor(["a", "b"])
    table.lookup(string_placeholder)

    with self.assertRaisesRegexp(ValueError, "Found table initializers"):

Instances