Skip to content

Commit 435112c

Browse files
Fix typing for RandomErasing
1 parent 229d852 commit 435112c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

torchvision/transforms/v2/_augment.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def __init__(
5858
p: float = 0.5,
5959
scale: Sequence[float] = (0.02, 0.33),
6060
ratio: Sequence[float] = (0.3, 3.3),
61-
value: float = 0.0,
61+
value: Union[float, Sequence[float]] = 0.0,
6262
inplace: bool = False,
6363
):
6464
super().__init__(p=p)

0 commit comments

Comments
 (0)