Format ctc_decode test correctly.

2020-05-28 00:02:04 -05:00 · 2020-05-28 00:02:04 -05:00 · f1487900c6
commit f1487900c6
parent a24a75aee1
1 changed files with 4 additions and 1 deletions
--- a/tensorflow/python/keras/backend_test.py
+++ b/tensorflow/python/keras/backend_test.py
@ -1771,7 +1771,10 @@ class TestCTC(test.TestCase):
        -3.777835    # output beam 1
    ], np.float32)[np.newaxis, :]

-    decode_truth = [np.array([1, 0, -1, -1, -1, -1, -1]), np.array([0, 1, 0, -1, -1, -1 ,-1])]
+    decode_truth = [
+        np.array([1, 0, -1, -1, -1, -1, -1]),
+        np.array([0, 1, 0, -1, -1, -1, -1])
+    ]
    beam_width = 2
    top_paths = 2