Skip to content

Commit ceb2cd9

Browse files
committed
Fixed mypy issue
1 parent 7d460af commit ceb2cd9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/prototype/transforms/functional/_geometry.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -535,7 +535,7 @@ def center_crop_bounding_box(
535535
format: features.BoundingBoxFormat,
536536
output_size: List[int],
537537
image_size: Tuple[int, int],
538-
):
538+
) -> torch.Tensor:
539539
crop_height, crop_width = _center_crop_parse_output_size(output_size)
540540
crop_top, crop_left = _center_crop_compute_crop_anchor(crop_height, crop_width, *image_size)
541541
return crop_bounding_box(bounding_box, format, top=crop_top, left=crop_left)

0 commit comments

Comments
 (0)