Add ValueError

Author: adityagandhamal

torchvision/ops/ciou_loss.py

@@ -72,5 +72,7 @@ def complete_box_iou_loss(
     elif reduction == "sum":
         loss = loss.sum()
     else:
-        raise ValueError(f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'")
+        raise ValueError(
+            f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'"
+        )
     return loss

torchvision/ops/diou_loss.py

@@ -58,7 +58,9 @@ def distance_box_iou_loss(
     elif reduction == "sum":
         loss = loss.sum()
     else:
-        raise ValueError(f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'")
+        raise ValueError(
+            f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'"
+        )
     return loss
 
 

torchvision/ops/focal_loss.py

@@ -36,7 +36,9 @@ def sigmoid_focal_loss(
     # Check reduction option
     modes = ["none", "mean", "sum"]
     if reduction not in modes:
-        raise ValueError(f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'")
+        raise ValueError(
+            f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'"
+        )
 
     if not torch.jit.is_scripting() and not torch.jit.is_tracing():
         _log_api_usage_once(sigmoid_focal_loss)

torchvision/ops/giou_loss.py

@@ -70,5 +70,7 @@ def generalized_box_iou_loss(
     elif reduction == "sum":
         loss = loss.sum()
     else:
-        raise ValueError(f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'")
+        raise ValueError(
+            f"Invalid Value for arg 'reduction': '{reduction} \n Supported reduction modes: 'none', 'mean', 'sum'"
+        )
     return loss