fix docs

pytorch · Aug 30, 2023 · da197bd · da197bd
1 parent 5da43b3
commit da197bd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gallery/transforms/plot_tv_tensors.py b/gallery/transforms/plot_tv_tensors.py
@@ -164,7 +164,7 @@
 # as a global config setting for the whole program, or as a context manager
 # (read its docs to learn more about caveats):
 
-with tv_tensors.set_return_type("tv_tensor"):
+with tv_tensors.set_return_type("TVTensor"):
     new_bboxes = bboxes + 3
 assert isinstance(new_bboxes, tv_tensors.BoundingBoxes)