Fix output type of upsampling (#801)

Fix output type of upsampling Co-authored-by: Ji Chen <jimchen90@devfair0160.h2.fair>
pytorch · Jul 20, 2020 · 937d52f · 937d52f
1 parent e82cc35
commit 937d52f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchaudio/models/_wavernn.py b/torchaudio/models/_wavernn.py
@@ -1,4 +1,4 @@
-from typing import List
+from typing import List, Tuple
 
 import torch
 from torch import Tensor
@@ -172,7 +172,7 @@ def __init__(self,
             up_layers.append(conv)
         self.upsample_layers = nn.Sequential(*up_layers)
 
-    def forward(self, specgram: Tensor) -> Tensor:
+    def forward(self, specgram: Tensor) -> Tuple[Tensor, Tensor]:
         r"""Pass the input through the _UpsampleNetwork layer.
 
         Args: