380b177c6339818c59d34db974a9690f1672f06b,opennmt/tests/inputter_test.py,InputterTest,_makeDataset,#InputterTest#Any#Any#Any#Any#Any#,152

Before Change

          self.assertIn(field, features)
          self.assertAllEqual(shape, features[field].get_shape().as_list())

    transformed = inputter.transform_data(next_element)
    with self.test_session() as sess:
      sess.run(tf.tables_initializer())
      sess.run(tf.global_variables_initializer())
      sess.run(iterator.initializer)
      return sess.run((next_element, transformed))

  @test_util.run_tf1_only
  def testWordEmbedder(self):

After Change

    if compat.is_tf2():
      iterator = None
      features = iter(dataset).next()
    else:
      iterator = dataset.make_initializable_iterator()
      features = iterator.get_next()

Instances