Fix format.

qlzh727 · Jun 25, 2020 · deb57c3 · deb57c3
1 parent 4b64158
commit deb57c3
Showing 1 changed file with 10 additions and 6 deletions.
diff --git 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
+    )