lint

Author: NicolasHug

test/test_tv_tensors.py

@@ -433,5 +433,10 @@ def test_return_type_input():
 
 
 def test_box_clamping_mode_default():
-    assert tv_tensors.BoundingBoxes([0., 0., 10., 10.], format="XYXY", canvas_size=(100, 100)).clamping_mode == "soft"
-    assert tv_tensors.BoundingBoxes([0., 0., 10., 10., 0.], format="XYWHR", canvas_size=(100, 100)).clamping_mode == "soft"
+    assert (
+        tv_tensors.BoundingBoxes([0.0, 0.0, 10.0, 10.0], format="XYXY", canvas_size=(100, 100)).clamping_mode == "soft"
+    )
+    assert (
+        tv_tensors.BoundingBoxes([0.0, 0.0, 10.0, 10.0, 0.0], format="XYWHR", canvas_size=(100, 100)).clamping_mode
+        == "soft"
+    )

torchvision/tv_tensors/_bounding_boxes.py

@@ -42,7 +42,11 @@ class BoundingBoxFormat(Enum):
 # this can be put into BoundingBoxFormat as staticmethod
 def is_rotated_bounding_format(format: Union[BoundingBoxFormat, str]) -> bool:
     if isinstance(format, BoundingBoxFormat):
-        return (format == BoundingBoxFormat.XYWHR or format == BoundingBoxFormat.CXCYWHR or format == BoundingBoxFormat.XYXYXYXY)
+        return (
+            format == BoundingBoxFormat.XYWHR
+            or format == BoundingBoxFormat.CXCYWHR
+            or format == BoundingBoxFormat.XYXYXYXY
+        )
     elif isinstance(format, str):
         return format in ("XYWHR", "CXCYWHR", "XYXYXYXY")
     else: