Fix flake8

BloodAxe · Nov 21, 2024 · f7a436d · f7a436d
1 parent e4caad5
commit f7a436d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pytorch_toolbelt/modules/interfaces.py b/pytorch_toolbelt/modules/interfaces.py
@@ -108,7 +108,8 @@ def __init__(self, input_spec: FeatureMapsSpecification):
     @abstractmethod
     def forward(
         self, feature_maps: List[Tensor], output_size: Union[Tuple[int, int], torch.Size, None] = None
-    ) -> Union[Tensor, Tuple[Tensor, ...], List[Tensor], Mapping[str, Tensor]]: ...
+    ) -> Union[Tensor, Tuple[Tensor, ...], List[Tensor], Mapping[str, Tensor]]:
+        ...
 
     @torch.jit.unused
     def apply_to_final_layer(self, func: Callable[[nn.Module], None]):