Fix formatting

Author: faizan-m

tensorflow_addons/image/dense_image_warp.py

@@ -52,6 +52,8 @@ def interpolate_bilinear(
     with tf.name_scope(name or "interpolate_bilinear"):
         grid = tf.convert_to_tensor(grid)
         query_points = tf.convert_to_tensor(query_points)
+        grid_shape = tf.shape(grid)
+        query_shape = tf.shape(query_points)
 
         tf.Assert(tf.equal(tf.rank(grid), 4), ["Grid must be 4D Tensor"])
         tf.Assert(

tensorflow_addons/image/tests/dense_image_warp_test.py

@@ -240,8 +240,8 @@ def test_size_exception():
     """Make sure it throws an exception for images that are too small."""
     shape = [1, 2, 1, 1]
     with pytest.raises(
-        tf.errors.InvalidArgumentError, match="Grid width must be at least 2."):
-      _check_interpolation_correctness(shape, "float32", "float32")
+          tf.errors.InvalidArgumentError, match="Grid width must be at least 2."):
+        _check_interpolation_correctness(shape, "float32", "float32")
 
 
 @pytest.mark.usefixtures("maybe_run_functions_eagerly")