fix ufmt format

pytorch · Sep 11, 2024 · 3233cf7 · 3233cf7
1 parent 4afe54a
commit 3233cf7
Showing 1 changed file with 1 addition and 6 deletions.
diff --git a/torchvision/tv_tensors/_dataset_wrapper.py b/torchvision/tv_tensors/_dataset_wrapper.py
@@ -384,12 +384,7 @@ def wrapper(idx, sample):
         coco_ann = dataset.coco.imgToAnns[image_id]
         if "masks" in target_keys:
             target["masks"] = tv_tensors.Mask(
-                torch.stack(
-                    [
-                        torch.from_numpy(dataset.coco.annToMask(ann))
-                        for ann in coco_ann
-                    ]
-                ),
+                torch.stack([torch.from_numpy(dataset.coco.annToMask(ann)) for ann in coco_ann]),
             )
 
         if "labels" in target_keys: