Run decoder tests in eager mode
@@ -62,7 +62,6 @@ class DecoderTest(tf.test.TestCase):
initial_state_fn=None,
num_sources=1,
dtype=tf.float32):
- with tf.Graph().as_default():
batch_size = 4
vocab_size = 10
time_dim = 5
@@ -92,7 +91,6 @@ class DecoderTest(tf.test.TestCase):
sequence_length,
state=initial_state,
training=True)
- self.evaluate(tf.compat.v1.global_variables_initializer())
self.assertEqual(outputs.dtype, dtype)
output_time_dim = tf.shape(outputs)[1]
if decoder.support_alignment_history and num_sources == 1:
