Code style pass

Author: Rocketknight1

tensorflow_addons/layers/tests/embedding_bag_test.py

@@ -100,7 +100,8 @@ def test_backward(input_shape, input_dim, dtype, indices_dtype, combiner, graph_
         expected_grads = tape.gradient(expected, [params, weights])
         # Gather returns sparse IndexedSlices so we have to sum them together.
         test_utils.assert_allclose_according_to_type(
-            tf.convert_to_tensor(expected_grads[0]), tf.convert_to_tensor(grads[0]),
+            tf.convert_to_tensor(expected_grads[0]),
+            tf.convert_to_tensor(grads[0]),
         )
         test_utils.assert_allclose_according_to_type(
             expected_grads[1],
@@ -116,4 +117,5 @@ def test_backward(input_shape, input_dim, dtype, indices_dtype, combiner, graph_
         expected_grads = tape.gradient(expected, [params])
         # Gather returns sparse IndexedSlices so we have to sum them together.
         test_utils.assert_allclose_according_to_type(
-            tf.convert_to_tensor(expected_grads[0]), tf.convert_to_tensor(grads[0]))
+            tf.convert_to_tensor(expected_grads[0]), tf.convert_to_tensor(grads[0])
+        )