fix BoundingBox.wrap_like (#7327)

e0eed2cb · Philip Meier · GitHub · 56d4078e · e0eed2cb
Unverified Commit e0eed2cb authored Feb 24, 2023 by Philip Meier Committed by GitHub Feb 24, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datapoints/_bounding_box.py torchvision/datapoints/_bounding_box.py +1 -1

No files found.
--- a/torchvision/datapoints/_bounding_box.py
+++ b/torchvision/datapoints/_bounding_box.py
@@ -87,7 +87,7 @@ class BoundingBox(Datapoint):

        """
        if isinstance(format, str):
-            format = BoundingBoxFormat.from_str(format.upper())
+            format = BoundingBoxFormat[format.upper()]

        return cls._wrap(
            tensor,