Skip to content

Commit ae6eb58

Browse files
NicolasHugfacebook-github-bot
authored andcommitted
[fbsync] fix BoundingBox.wrap_like (#7327)
Reviewed By: vmoens Differential Revision: D44416543 fbshipit-source-id: a7b43807d30dc57958249b792913a3a44a6e204f
1 parent fdfe9e0 commit ae6eb58

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/datapoints/_bounding_box.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def wrap_like(
8787
8888
"""
8989
if isinstance(format, str):
90-
format = BoundingBoxFormat.from_str(format.upper())
90+
format = BoundingBoxFormat[format.upper()]
9191

9292
return cls._wrap(
9393
tensor,

0 commit comments

Comments
 (0)