format code with ufmt

Author: AndreiMoraru123

test/test_utils.py

@@ -164,27 +164,30 @@ def test_draw_boxes_with_coloured_label_backgrounds():
     )
     expected = torch.as_tensor(np.array(Image.open(path))).permute(2, 0, 1)
     assert_equal(result, expected)
-    
+
 
 @pytest.mark.skipif(PILLOW_VERSION < (10, 1), reason="The reference image is only valid for PIL >= 10.1")
 def test_draw_boxes_with_coloured_label_text_boxes():
     img = torch.full((3, 100, 100), 255, dtype=torch.uint8)
     labels = ["a", "b", "c", "d"]
     colors = ["green", "#FF00FF", (0, 255, 0), "red"]
     label_colors = ["green", "red", (0, 255, 0), "#FF00FF"]
-    label_background_colors = ["white", "black", "yellow", "blue"] 
+    label_background_colors = ["white", "black", "yellow", "blue"]
     result = utils.draw_bounding_boxes(
-        img, 
-        boxes, 
-        labels=labels, 
-        colors=colors, 
-        fill=True, 
+        img,
+        boxes,
+        labels=labels,
+        colors=colors,
+        fill=True,
         label_colors=label_colors,
         label_background_colors=label_background_colors,
-        fill_labels=True
+        fill_labels=True,
     )
     path = os.path.join(
-        os.path.dirname(os.path.abspath(__file__)), "assets", "fakedata", "draw_boxes_different_label_background_colors.png"
+        os.path.dirname(os.path.abspath(__file__)),
+        "assets",
+        "fakedata",
+        "draw_boxes_different_label_background_colors.png",
     )
     expected = torch.as_tensor(np.array(Image.open(path))).permute(2, 0, 1)
     assert_equal(result, expected)

torchvision/utils.py

@@ -8,8 +8,7 @@
 
 import numpy as np
 import torch
-from PIL import Image, ImageColor, ImageDraw, ImageFont
-from PIL import __version__ as PILLOW_VERSION_STRING
+from PIL import __version__ as PILLOW_VERSION_STRING, Image, ImageColor, ImageDraw, ImageFont
 
 __all__ = [
     "_Image_fromarray",
@@ -366,7 +365,7 @@ def draw_bounding_boxes(
         label_colors = colors.copy()  # type: ignore[assignment]
 
     if fill_labels:
-        label_background_colors = _parse_colors(label_background_colors, num_objects=num_boxes) if label_background_colors else colors.copy() # type: ignore[assignment]
+        label_background_colors = _parse_colors(label_background_colors, num_objects=num_boxes) if label_background_colors else colors.copy()  # type: ignore[assignment]
     else:
         label_background_colors = colors.copy()  # type: ignore[assignment]