Skip to content

Commit e0eed2c

Browse files
authored
fix BoundingBox.wrap_like (#7327)
1 parent 56d4078 commit e0eed2c

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)