Skip to content

GitLab

Unverified Commit b1a704d7 authored by Katherine Wu's avatar Katherine Wu Committed by GitHub
Browse files

Fix transformer test (#4606)

parent c39a88c0
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
official/transformer/model/beam_search_test.py
View file @ b1a704d7
......@@ -33,8 +33,8 @@ class BeamSearchHelperTests(tf.test.TestCase):
self.assertAllEqual([7, 3, 4, 2, 5], shape)
def test_shape_list(self):
y = tf.constant(4.0)
x = tf.ones([7, tf.to_int32(tf.sqrt(y)), 2, 5])
y = tf.placeholder(dtype=tf.int32, shape=[])
x = tf.ones([7, y, 2, 5])
shape = beam_search._shape_list(x)
self.assertIsInstance(shape[0], int)
self.assertIsInstance(shape[1], tf.Tensor)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment