fix error message if v2 transform is not JIT scriptable (#7196)

pmeier · web-flow · commit 378a3274b178 · 2023-02-08T22:05:06.000+01:00
diff --git a/torchvision/prototype/transforms/_transform.py b/torchvision/prototype/transforms/_transform.py
@@ -130,7 +130,7 @@ def __prepare_scriptable__(self) -> nn.Module:
         # is around.
         if self._v1_transform_cls is None:
             raise RuntimeError(
-                f"Transform {type(self.__name__)} cannot be JIT scripted. "
+                f"Transform {type(self).__name__} cannot be JIT scripted. "
                 f"This is only support for backward compatibility with transforms which already in v1."
                 f"For torchscript support (on tensors only), you can use the functional API instead."
             )

Original file line number	Diff line number	Diff line change
`@@ -130,7 +130,7 @@ def __prepare_scriptable__(self) -> nn.Module:`
`130`	`130`	`# is around.`
`131`	`131`	`if self._v1_transform_cls is None:`
`132`	`132`	`raise RuntimeError(`
`133`		`- f"Transform {type(self.__name__)} cannot be JIT scripted. "`
	`133`	`+ f"Transform {type(self).__name__} cannot be JIT scripted. "`
`134`	`134`	`f"This is only support for backward compatibility with transforms which already in v1."`
`135`	`135`	`f"For torchscript support (on tensors only), you can use the functional API instead."`
`136`	`136`	`)`