diff --git a/torchtext/transforms.py b/torchtext/transforms.py index 99cff3fe12..37bd3706cb 100644 --- a/torchtext/transforms.py +++ b/torchtext/transforms.py @@ -111,7 +111,7 @@ class ToTensor(Module): :type dtype: :class:`torch.dtype` """ - def __init__(self, padding_value: Optional[int] = None, dtype: Optional[torch.dtype] = torch.long) -> None: + def __init__(self, padding_value: Optional[int] = None, dtype: torch.dtype = torch.long) -> None: super().__init__() self.padding_value = padding_value self.dtype = dtype