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