From 97bf1778105c9422dc06fc52c962d29f74894b12 Mon Sep 17 00:00:00 2001
From: vfdev <vfdev.5@gmail.com>
Date: Fri, 10 Nov 2023 16:25:02 +0100
Subject: [PATCH] Update torchvision/tv_tensors/__init__.py

Co-authored-by: Nicolas Hug <nh.nicolas.hug@gmail.com>
---
 torchvision/tv_tensors/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/torchvision/tv_tensors/__init__.py b/torchvision/tv_tensors/__init__.py
index 2e58d9d4c6a..441502df2f0 100644
--- a/torchvision/tv_tensors/__init__.py
+++ b/torchvision/tv_tensors/__init__.py
@@ -10,7 +10,7 @@
 
 # TODO: Fix this. We skip this method as it leads to
 # RecursionError: maximum recursion depth exceeded while calling a Python object
-# Keeping it here, leads to graph breaks between multiple functional ops instead of having a single graph
+# Until `disable` is removed, there will be graph breaks after all calls to functional transforms
 @torch.compiler.disable
 def wrap(wrappee, *, like, **kwargs):
     """[BETA] Convert a :class:`torch.Tensor` (``wrappee``) into the same :class:`~torchvision.tv_tensors.TVTensor` subclass as ``like``.