From 65dda6ca4883db57abac26efb1bbfc9a4d0a2a89 Mon Sep 17 00:00:00 2001 From: Elijah Rippeth Date: Fri, 7 Jan 2022 07:45:58 -0500 Subject: [PATCH] remove optionality of dtype in `ToTensor` --- torchtext/transforms.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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