fix v2.Lambda

pytorch · May 8, 2023 · 93c2d45 · 93c2d45
1 parent 0370134
commit 93c2d45
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/torchvision/transforms/v2/_misc.py b/torchvision/transforms/v2/_misc.py
@@ -32,10 +32,12 @@ class Lambda(Transform):
         lambd (function): Lambda/function to be used for transform.
     """
 
+    _transformed_types = (object,)
+
     def __init__(self, lambd: Callable[[Any], Any], *types: Type):
         super().__init__()
         self.lambd = lambd
-        self.types = types or (object,)
+        self.types = types or self._transformed_types
 
     def _transform(self, inpt: Any, params: Dict[str, Any]) -> Any:
         if isinstance(inpt, self.types):