Removed return typing hint that causes an error in Python 3.7.

aehrc · Jul 29, 2022 · 7765670 · 7765670
1 parent 13d450a
commit 7765670
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transmodal/network/cvt.py b/transmodal/network/cvt.py
@@ -74,7 +74,7 @@ def __init__(
             for n, p in self.cvt.named_parameters():
                 p.requires_grad = False
 
-    def forward(self, images: torch.FloatTensor) -> Union[dict[str, Tensor], dict[str, Union[Tensor, Any]]]:
+    def forward(self, images: torch.FloatTensor):
         """
         Forward propagation.