From deb57c329c92e844ad2eb2a9381f02f1526d009a Mon Sep 17 00:00:00 2001 From: qlzh727 Date: Wed, 24 Jun 2020 21:04:01 -0700 Subject: [PATCH] Fix format. --- tensorflow_addons/rnn/tests/cell_test.py | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 deletions(-) diff --git a/tensorflow_addons/rnn/tests/cell_test.py b/tensorflow_addons/rnn/tests/cell_test.py index a19a00356f..be3fd13d5e 100644 --- a/tensorflow_addons/rnn/tests/cell_test.py +++ b/tensorflow_addons/rnn/tests/cell_test.py @@ -585,13 +585,17 @@ def _run_cell(cell_fn, **kwargs): implementation=2, ) expected_output = np.asarray( - [[0.417551, 0.417551, 0.417551, 0.417551, 0.417551], - [0.417551, 0.417551, 0.417551, 0.417551, 0.417551], - [0.417551, 0.417551, 0.417551, 0.417551, 0.417551], - [0., 0., 0., 0., 0.]], - dtype=np.float32 + [ + [0.417551, 0.417551, 0.417551, 0.417551, 0.417551], + [0.417551, 0.417551, 0.417551, 0.417551, 0.417551], + [0.417551, 0.417551, 0.417551, 0.417551, 0.417551], + [0.0, 0.0, 0.0, 0.0, 0.0], + ], + dtype=np.float32, ) np.testing.assert_allclose( first_implementation_output, second_implementation_output ) - np.testing.assert_allclose(first_implementation_output, expected_output, rtol=1e-6, atol=1e-6) + np.testing.assert_allclose( + first_implementation_output, expected_output, rtol=1e-6, atol=1e-6 + )