Add ValueError

pytorch · Oct 1, 2022 · e178f36 · e178f36
1 parent 2d7466f
commit e178f36
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 4 deletions.
diff --git a/torchvision/ops/ciou_loss.py b/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
diff --git a/torchvision/ops/diou_loss.py b/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
 
 

diff --git a/torchvision/ops/focal_loss.py b/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)

diff --git a/torchvision/ops/giou_loss.py b/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