Update torchvision/tv_tensors/__init__.py

Co-authored-by: Nicolas Hug <[email protected]>

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``.