Format code

Author: fsx950223

tensorflow_addons/losses/giou_loss.py

@@ -26,9 +26,9 @@
 class GIOULoss(tf.keras.losses.Loss):
     """Implements the GIOU loss function.
 
-    GIOU loss was first introduced in the Generalized Intersection over Union paper
-    (https://giou.stanford.edu/GIoU.pdf). GIOU is a enhance for model which use IOU
-    in object detection.
+    GIOU loss was first introduced in the Generalized Intersection over Union
+    paper (https://giou.stanford.edu/GIoU.pdf). GIOU is a enhance for model
+    which use IOU in object detection.
 
     Usage:
 
@@ -73,7 +73,7 @@ def giou_loss(y_true, y_pred):
     Args
         y_true: true targets tensor.
         y_pred: predictions tensor.
-    
+
     Returns:
         GIOU loss float `Tensor`.
     """

tensorflow_addons/losses/giou_loss_test.py

@@ -25,6 +25,7 @@
 
 @test_utils.run_all_in_graph_and_eager_modes
 class GIOULossTest(tf.test.TestCase):
+    """GIOU test class"""
     def test_config(self):
         gl_obj = GIOULoss(
             reduction=tf.keras.losses.Reduction.NONE, name='giou_loss')