Reformatted

Lightning-Universe · Aug 19, 2021 · 66c8111 · 66c8111
1 parent 191e865
commit 66c8111
Showing 1 changed file with 2 additions and 6 deletions.
diff --git a/pl_bolts/models/detection/yolo/yolo_module.py b/pl_bolts/models/detection/yolo/yolo_module.py
@@ -354,16 +354,12 @@ def _validate_batch(
             if not isinstance(boxes, Tensor):
                 raise ValueError(f"Expected target boxes to be of type Tensor, got {type(boxes)}.")
             if (len(boxes.shape) != 2) or (boxes.shape[-1] != 4):
-                raise ValueError(
-                    f"Expected target boxes to be tensors of shape [N, 4], got {list(boxes.shape)}."
-                )
+                raise ValueError(f"Expected target boxes to be tensors of shape [N, 4], got {list(boxes.shape)}.")
             labels = target["labels"]
             if not isinstance(labels, Tensor):
                 raise ValueError(f"Expected target labels to be of type Tensor, got {type(labels)}.")
             if len(labels.shape) != 1:
-                raise ValueError(
-                    f"Expected target labels to be tensors of shape [N], got {list(labels.shape)}."
-                )
+                raise ValueError(f"Expected target labels to be tensors of shape [N], got {list(labels.shape)}.")
 
         images = torch.stack(images)
         return images, targets