fix error message

Author: pmeier

test/test_transforms_v2_refactored.py

@@ -5070,7 +5070,7 @@ def test_functional_and_transform(self, make_input, fn):
             assert output.data_ptr() == input.data_ptr()
 
     def test_functional_error(self):
-        with pytest.raises(TypeError, match="Input can either be a numpy array or a PIL image"):
+        with pytest.raises(TypeError, match="Input can either be a pure Tensor, a numpy array, or a PIL image"):
             F.to_image(object())
 
 

torchvision/transforms/v2/functional/_type_conversion.py

@@ -17,7 +17,9 @@ def to_image(inpt: Union[torch.Tensor, PIL.Image.Image, np.ndarray]) -> tv_tenso
     elif isinstance(inpt, torch.Tensor):
         output = inpt
     else:
-        raise TypeError(f"Input can either be a numpy array or a PIL image, but got {type(inpt)} instead.")
+        raise TypeError(
+            f"Input can either be a pure Tensor, a numpy array, or a PIL image, but got {type(inpt)} instead."
+        )
     return tv_tensors.Image(output)